POJ - 3294 Life Forms

传送门：POJ - 3294   (类似题目POJ - 3450)

题意：给你n个字符串，求在不小于k/2个字符串中出现过的最长子串。

题解：首先将n个字符串连起来，中间用不一样且没出现过的字符连起来（这个地方让我wa了好几发！！），然后二分字串的长度，将后缀分成若干组，判断每组的后缀是否在不小于k个的原串中出现过，数组sum记录每个最长字符串在原串中出现的位置sa[i]。

#include<cstdio>
#include<algorithm>
#include<queue>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;

const int maxn = 1000100;
int wa[maxn],wb[maxn],wsf[maxn],wv[maxn],sa[maxn];
int rnk[maxn],height[maxn],sum[maxn];
char s[maxn];
int r[maxn],n,pos=0,len[110],vis[110],ans=0,L,be[maxn];

//sa:字典序中排第i位的起始位置在str中第sa[i]  sa[1~n]为有效值

//rnk:就是str第i个位置的后缀是在字典序排第几 rnk[0~n-1]为有效值

//height：字典序排i和i-1的后缀的最长公共前缀  height[2~n]为有效值，第二个到最后一个

int cmp(int *r,int a,int b,int k)
{
    return r[a]==r[b]&&r[a+k]==r[b+k];
}

void getsa(int *r,int *sa,int n,int m)//n为添加0后的总长
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0; i<m; i++)  wsf[i]=0;
    for(i=0; i<=n; i++)  wsf[x[i]=r[i]]++;
    for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];
    for(i=n; i>=0; i--)  sa[--wsf[x[i]]]=i;
    p=1;
    j=1;
    for(; p<=n; j*=2,m=p){
        for(p=0,i=n+1-j; i<=n; i++)  y[p++]=i;
        for(i=0; i<=n; i++)  if(sa[i]>=j)  y[p++]=sa[i]-j;
        for(i=0; i<=n; i++)  wv[i]=x[y[i]];
        for(i=0; i<m; i++)  wsf[i]=0;
        for(i=0; i<=n; i++)  wsf[wv[i]]++;
        for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];
        for(i=n; i>=0; i--)  sa[--wsf[wv[i]]]=y[i];
        swap(x,y);
        x[sa[0]]=0;
        for(p=1,i=1; i<=n; i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)? p-1:p++;
    }
}

void getheight(int *r,int n)//n为添加0后的总长
{
    int i,j,k=0;
    for(i=1; i<=n; i++)  rnk[sa[i]]=i;
    for(i=0; i<n; i++){
        if(k)
            k--;
        else
            k=0;
        j=sa[rnk[i]-1];
        while(r[i+k]==r[j+k])
            k++;
        height[rnk[i]]=k;
    }
}

bool check(int k)
{
    int cnt=0,p=0;
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=L;i++){
        if(height[i]>=k){
            if(!vis[be[sa[i]]]) cnt++;
            vis[be[sa[i]]]=1;
            if(!vis[be[sa[i-1]]]) cnt++;
            vis[be[sa[i-1]]]=1;
        }
        else{
            if(cnt>n/2) sum[p++]=sa[i-1];
            cnt=0;
            memset(vis,0,sizeof(vis));
        }
    }
    if(cnt>n/2) sum[p++]=sa[L];
    cnt=0;
    if(p) ans=p;
    if(p) return 1;
    return 0;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t=0;
    while(cin>>n&&n){
        L=0;
        for(int i=1;i<=n;i++){
            cin>>s+L;
            int p=strlen(s+L);
            for(int j=0;j<p;j++){
                r[j+L]=s[j+L]-'A'+1;
                be[j+L]=i;
            }
            L+=p;
            len[i]=L;
            if(i!=n){
                r[L]=100+i;
                be[L]=i;
                L++;
            }
        }
        r[L]=0;
        getsa(r,sa,L,300);
        getheight(r,L);
        int l=1,r=L;
        ans=0;
        while(l<=r){
            int mid=l+r>>1;
            if(check(mid)){
                l=mid+1;
            }
            else r=mid-1;
        }
        if(t) cout<<endl;
        t=1;
        if(l==1) cout<<"?"<<endl;
        for(int i=0;i<ans;i++){
            for(int j=sum[i];j<sum[i]+l-1;j++){
                cout<<s[j];
            }
            cout<<endl;
        }
    }
    return 0;
}