[POJ3294]Life Forms（后缀数组+二分）

题目描述

传送门

题解

将 n 个字符串连起来，中间用不相同的且没有出现在字符串中的字符隔开，求sa和height。然后二分答案，将后缀分成若干组，每一组的height都不小于mid，判断每组后缀是否出现在不小于 k=n/2+1 个的原串中。
按字典序输出的话就记一下位置然后根据sa中的位置输出就行了。

注意：
①n=1的时候要特判
②每一个height都不能超过它本身表示的字符串的长度

代码

#include<iostream>#include<cstring>#include<cstdio>using namespace std;#define N 200005int T,k,n,m,la,Max;char a[N],s[N];int *x,*y,X[N],Y[N],c[N],sa[N],height[N],rank[N];int str[N],is_end[N],flag[N],ans[N];void clear(){    k=n=m=la=Max=0;    memset(a,0,sizeof(a));memset(s,0,sizeof(s));    memset(X,0,sizeof(X));memset(Y,0,sizeof(Y));memset(c,0,sizeof(c));    memset(sa,0,sizeof(sa));memset(height,0,sizeof(height));memset(rank,0,sizeof(rank));    memset(str,0,sizeof(str));memset(is_end,0,sizeof(is_end));memset(flag,0,sizeof(flag));memset(ans,0,sizeof(ans));}void build_sa(){    m=200;    x=X,y=Y;    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=1;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]]&&((sa[i-1]+k<n?y[sa[i-1]+k]:-1)==(sa[i]+k<n?y[sa[i]+k]:-1))?p-1:p++;        if (p>n) break;        m=p;    }}void build_height(){    for (int i=0;i<n;++i) rank[sa[i]]=i;    int k=0;height[0]=0;    for (int i=0;i<n;++i)    {        if (!rank[i]) continue;        if (k) --k;        int j=sa[rank[i]-1];        while (i+k<n&&j+k<n&&s[i+k]==s[j+k]) ++k;        height[rank[i]]=k;    }}bool check(int mid){    bool pd=false;    int last=T-1;    for (int i=T;i<=n;++i)    {        if (height[i]<mid||i==n)        {            int cnt=0;            for (int j=last;j<i;++j)            {                if (!flag[str[sa[j]]]) ++cnt;                ++flag[str[sa[j]]];            }            if (cnt>=k)            {                if (!pd) pd=true,ans[0]=0;                ans[++ans[0]]=last;            }            for (int j=last;j<i;++j)                --flag[str[sa[j]]];            last=i;        }    }    return pd;}int find(){    int l=0,r=Max,mid,ans=0;    while (l<=r)    {        mid=(l+r)>>1;        if (check(mid)) ans=mid,l=mid+1;        else r=mid-1;    }    return ans;}int main(){    while (~scanf("%d\n",&T))    {        if (!T) break;        clear();        k=T/2+1;        for (int i=1;i<=T;++i)        {            gets(a);la=strlen(a);            if (k==1)            {                if (la>Max) ans[0]=1,ans[1]=n+1;                else if (la==Max) ans[++ans[0]]=n+1;            }            Max=max(Max,la);            for (int j=0;j<la;++j)                s[n++]=a[j],str[n-1]=i;            if (i!=T) s[n++]='$',str[n-1]=i;            is_end[i]=n-1;        }        if (k==1)        {            for (int i=1;i<=ans[0];++i)            {                for (int j=0;j<Max;++j)                    putchar(s[sa[i]+j]);putchar('\n');                puts("");            }            continue;        }        is_end[T]=n;        build_sa();        build_height();        for (int i=T-1;i<n;++i)            height[i]=min(height[i],is_end[str[sa[i]]]-sa[i]);        int len=find();        if (!len)        {            puts("?");            puts("");            continue;        }        for (int i=1;i<=ans[0];++i)        {            for (int j=0;j<len;++j)                putchar(s[sa[ans[i]]+j]);putchar('\n');        }        puts("");    }}