poj 3294 Life Forms

后缀数组的题目，把后缀连接起来，这个还是先二分答案，然后选取一段连续的height值，判断这些height代表的后缀有没有覆盖一半以上的字符串。

得出答案的长度之后还要在枚举连续的heigh，判断有没有答案，有的话标记其中一个。

最后再按照sa输出答案。这样就可以保证字典序。

 

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e5+10000;
int r[maxn];
char a[1111];
int *rank,height[maxn],sa[maxn];
int wx[maxn],wy[maxn],cnt[maxn];
int n,m;
inline bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int n,int m)
{
    r[n+1]=0;
    int i,l,p,*x=wx,*y=wy,*t;
    memset(cnt,0,sizeof(int)*(m+1));
    for(int i=1;i<=n;i++) cnt[x[i]=r[i]]++;
    for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for(int i=n;i>=1;i--) sa[cnt[x[i]]--]=i;
    for(l=1,p=1;p<n;l<<=1,m=p)
    {
        for(p=1,i=n-l+1;i<=n;i++) y[p++]=i;
        for(i=1;i<=n;i++)
        if(sa[i]>l)
        y[p++]=sa[i]-l;
        memset(cnt,0,sizeof(int)*(m+1));
        for(i=1;i<=n;i++) cnt[x[i]]++;
        for(i=1;i<=m;i++) cnt[i]+=cnt[i-1];
        for(i=n;i>=1;i--) sa[cnt[x[y[i]]]--]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[1]]=1,i=2;i<=n;i++)
        x[sa[i]]=cmp(y,sa[i-1],sa[i],l)?p:++p;
    }
    rank=x;
    int j,k=0;
	for(i=1;i<=n;height[rank[i++]]=k)
	for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
	return;
}

bool fla[1111];
int c[maxn];
bool ans[maxn];
bool chk(int ret)
{
    for(int i=2;i<=n;i++)
    if(height[i]>=ret)
    {
        int j=i+1;
        while(j<=n&&height[j]>=ret) j++;
        j--;
        memset(fla,0,sizeof(fla));
        for(int k=i-1;k<=j;k++)
        fla[c[sa[k]]]=1;
        int cnt=0;
        for(int k=1;k<=m;k++)
        cnt+=fla[k];
        if(cnt>m/2) return true;
        i=j;
    }
    return false;
}

int main()
{
//    freopen("in.txt","r",stdin);
    int ret;
    int tt=0;
    while(scanf("%d",&m),m)
    {
        if(tt++) printf("\n");
        n=0;
        for(int i=1;i<=m;i++)
        {
            scanf("%s",a+1);
            ret=strlen(a+1);
            for(int j=1;j<=ret;j++)
            r[n+j]=a[j];
            for(int j=n+1;j<=n+ret;j++)
            c[j]=i;
            n+=ret;
            r[++n]=570+i;
        }
        da(r,n,680);
        int st=0,ed=1111,mid;
        while(st<ed)
        {
            mid=st+ed+1>>1;
            if(chk(mid)) st=mid;
            else ed=mid-1;
        }
        if(m==1)
        {
            for(int i=1;i<n;i++)
            printf("%c",r[i]);
            printf("\n");
        }
        else if(st==0)
        printf("?\n");
        else
        {
            memset(ans,0,sizeof(ans));
            for(int i=1;i<=n;i++)
            if(height[i]>=st)
            {
                int j=i+1;
                while(height[j]>=st) j++;
                j--;
                memset(fla,0,sizeof(fla));
                for(int k=i-1;k<=j;k++)
                fla[c[sa[k]]]=1;
                int cnt=0;
                for(int k=1;k<=m;k++)
                cnt+=fla[k];
                if(cnt>m/2)
                {
                    ans[sa[i-1]]=1;
                }
                i=j;
            }
            for(int i=1;i<=n;i++)
            if(ans[sa[i]])
            {
                for(int j=1;j<=st;j++)
                printf("%c",r[j+sa[i]-1]);
                printf("\n");
            }
        }
    }
    return 0;
}