UVA 11107 Life Forms(后缀数组)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100000+1000;
struct SuffixArray{
    int s[maxn];
    int sa[maxn];
    int rank[maxn];
    int height[maxn];
    int t[maxn],t2[maxn],c[maxn];
    int n;
    void clear() {n=0;memset(sa,0,sizeof(sa));}
    void build_sa(int m)
    {
        int *x=t,*y=t2;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[i]=s[i]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
        for(int k=1;k<=n;k<<=1)
        {
            int p=0;
            for(int i=n-k;i<n;i++) y[p++]=i;
            for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
            for(int i=0;i<m;i++) c[i]=0;
            for(int i=0;i<n;i++) c[x[y[i]]]++;
            for(int i=0;i<m;i++) c[i]+=c[i-1];
            for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
            swap(x,y);
            p=1;x[sa[0]]=0;
            for(int i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
            m=p;
        }
    }
    
    void build_height()
    {
        int k=0;
        for(int i=0;i<n;i++) rank[sa[i]]=i;
        for(int i=0;i<n;i++)
        {
            if(k) k--;
            int j=sa[rank[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[rank[i]]=k;
        }
    }
};

int n,maxlen;
char word[1005];
int idx[maxn];
int flag[105];
#define ss(x) scanf("%d",&x)
SuffixArray sa;
bool good(int L,int R)
{
    if(R-L<=n/2) return false;
    memset(flag,0,sizeof(flag));
    int cnt=0;
    for(int i=L;i<R;i++)
    {
        int x=idx[sa.sa[i]];//因为height定义的是sa[i-1],sa[i]的LCP，所以L与R其实是sa[L]~sa[R]的范围
        if(x!=n&&!flag[x]) {flag[x]=1;cnt++;}
    }
    return cnt>n/2;
}
void print_sub(int L,int R)
{
    for(int i=L;i<R;i++)
        printf("%c",sa.s[i]-1+'a');
    printf("\n");
}
bool print_solution(int len,bool print)
{
    int L=0;
    for(int R=1;R<=sa.n;R++)
    {
        if(R==sa.n||sa.height[R]<len)
        {
            if(good(L,R)){
                if(print) print_sub(sa.sa[L],sa.sa[L]+len);else return true;
            }
            L=R;
        }
    }
    return false;
}
void solve(int maxlen)
{
    if(!print_solution(1,false)) printf("?\n");
    else{
        int L=1,R=maxlen,M;
        while(L<R)
        {
            M=L+(R-L+1)/2;
            if(print_solution(M,false)) L=M;
            else R=M-1;
        }
        print_solution(L,true);
    }
}
void add(int ch,int i)
{
    idx[sa.n]=i;
    sa.s[sa.n++]=ch;
}
int main()
{
    int kase=0;
    while(true)
    {
        ss(n);if(n==0) break;
        maxlen=0;
        sa.clear();
        if(kase++>0) printf("\n");
        for(int i=0;i<n;i++)
        {
            scanf("%s",word);
            maxlen=max(0,(int)strlen(word));
            for(int j=0;word[j];j++)
                add(word[j]-'a'+1,i);
            add(100+i,n);
        }
        add(0,n);
        if(n==1) printf("%s\n",word);
        else{
            sa.build_sa(100+n);
            sa.build_height();
            solve(maxlen);
        }
    }
    return 0;
}