【POJ3294】 Life Forms(SA)

...又是TLE，对于单组数据肯定TLE不了，问题是多组的时候就呵呵了...

按height分组去搞，然后判一下是否不属于同一个串...

const maxn=;
var
 x,y,rank,sa,c,col,h,rec:array[..maxn] of longint;
 pd:array[..maxn] of boolean;
 s:ansistring;
 n,k,maxlen,cnta:longint;

function max(x,y:longint):longint; begin if x>y then exit(x) else exit(y); end;
function min(x,y:longint):longint; begin if x<y then exit(x) else exit(y); end;
procedure swap(var x,y:longint);var tmp:longint; begin tmp:=x;x:=y;y:=tmp; end;

procedure make;
var i,p,tot:longint;
begin
 p:=;
 while p<n do
  begin
   fillchar(c,sizeof(c),);
   for i:=  to n-p do y[i]:=rank[i+p];
   for i:= n-p+ to n do y[i]:=;
   for i:=  to n do inc(c[y[i]]);
   for i:=  to n do inc(c[i],c[i-]);
   for i:=  to n do
    begin
     sa[c[y[i]]]:=i;
     dec(c[y[i]]);
    end;
   fillchar(c,sizeof(c),);
   for i:=  to n do x[i]:=rank[i];
   for i:=  to n do inc(c[x[i]]);
   for i:=  to n do inc(c[i],c[i-]);
   for i:= n downto  do
    begin
     y[sa[i]]:=c[x[sa[i]]];
     dec(c[x[sa[i]]]);
    end;
   for i:=  to n do sa[y[i]]:=i;
   tot:=;
   rank[sa[]]:=;
   for i:=  to n do
    begin
     if (x[sa[i]]<>x[sa[i-]]) or (x[sa[i]+p]<>x[sa[i-]+p]) then inc(tot);
     rank[sa[i]]:=tot;
    end;
   p:=p<<;
  end;
 for i:=  to n do sa[rank[i]]:=i;
end;

procedure makeh;
var i,p,j:longint;
begin
 p:=; h[]:=;
 for i:=  to n do
  begin
   p:=max(p-,);
   if rank[i]= then continue;
   j:=sa[rank[i]-];
   while (j+p<=n) and (i+p<=n) and (s[i+p]=s[j+p]) do inc(p);
   h[rank[i]]:=p;
  end;
end;

procedure init;
var i,j,tot:longint;
 s1:ansistring;
begin
 k:=n>>;
 readln(s);
 tot:=;
 maxlen:=length(s);
 for i:=  to maxlen do col[i]:=;
 for i:=  to n do
  begin
   readln(s1);
   s:=s+'#';
   maxlen:=max(length(s1),maxlen);
   inc(tot);
   for j:= length(s)+ to length(s)+length(s1) do col[j]:=tot;
   s:=s+s1;
  end;
 n:=length(s);
 fillchar(c,sizeof(c),);
 for i:=  to n do x[i]:=ord(s[i]);
 for i:=  to n do inc(c[x[i]]);
 for i:=  to  do inc(c[i],c[i-]);
 for i:=  to n do
  begin
   sa[c[x[i]]]:=i;
   dec(c[x[i]]);
  end;
 tot:=;
 rank[sa[]]:=;
 for i:=  to n do
  begin
   if x[sa[i]]<>x[sa[i-]] then inc(tot);
   rank[sa[i]]:=tot;
  end;
 make; makeh;
end;

function check(x:longint):boolean;
var i,j,tot,cnt:longint;
begin
 fillchar(pd,sizeof(pd),false);
 pd[]:=true; tot:=; cnt:=;
 if not pd[sa[]] then inc(tot);
 pd[sa[]]:=true;
 for i:=  to n do
  begin
   if h[i]<x then
    begin
     if tot>k then
      begin
       inc(cnt);
       rec[cnt]:=i-;
      end;
     tot:=;
     fillchar(pd,sizeof(pd),false);
     pd[]:=true;
    end;
   if not pd[col[sa[i]]] then inc(tot);
   pd[col[sa[i]]]:=true;
  end;
 if tot>k then
  begin
   inc(cnt);
   rec[cnt]:=i-;
  end;
 if cnt> then
  begin
   cnta:=cnt;
   exit(true);
  end;
 exit(false);
end;

procedure qsort(l,r:longint);
var i,j,mid,tmp:longint;
begin
 if l>r then exit;
 i:=l; j:=r;
 mid:=rec[l+random(r-l+)];
 repeat
  while rec[i]<mid do inc(i);
  while rec[j]>mid do dec(j);
  if i<=j then
   begin
    swap(i,j);
    inc(i); dec(j);
   end;
 until i>j;
 qsort(i,r); qsort(l,j);
end;

procedure solve;
var l,r,mid,tot,ans,i,j:longint;
begin
 l:=; r:=maxlen;  ans:=;
 while l<=r do
  begin
   mid:=(l+r)>>;
   if check(mid) then
    begin
     ans:=mid;
     l:=mid+;
    end
   else r:=mid-;
  end;
 fillchar(pd,sizeof(pd),false);
 if ans= then
  begin
   writeln('?');
   exit;
  end;
 qsort(,cnta);
 for i:=  to cnta do writeln(copy(s,sa[rec[i]],ans));
end;

Begin
 readln(n);
 while n<> do
  begin
   init;
   solve;
   writeln;
   readln(n);
  end;
End.