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

给n个字符串，求最长的多于n/2个字符串的公共子串。

依然是二分判定+height分组。

• 把这n个字符串连接，中间用不同字符隔开，跑后缀数组计算出height；
• 二分要求的子串长度，判断是否满足：height分组，统计一个组不同的字符串个数是否大于n/2；
• 最后输出方案，根据二分得出的子串长度的结果，直接再遍历一遍height，因为这儿是有序的后缀所以找到一个就直接输出。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 111000

int wa[MAXN],wb[MAXN],wv[MAXN],ws[MAXN];
int cmp(int *r,int a,int b,int l){
    return r[a]==r[b] && r[a+l]==r[b+l];
}
int sa[MAXN],rank[MAXN],height[MAXN];
void SA(int *r,int n,int m){
    int *x=wa,*y=wb;

    for(int i=0; i<m; ++i) ws[i]=0;
    for(int i=0; i<n; ++i) ++ws[x[i]=r[i]];
    for(int i=1; i<m; ++i) ws[i]+=ws[i-1];
    for(int i=n-1; i>=0; --i) sa[--ws[x[i]]]=i;

    int p=1;
    for(int j=1; p<n; j<<=1,m=p){
        p=0;
        for(int i=n-j; i<n; ++i) y[p++]=i;
        for(int i=0; i<n; ++i) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(int i=0; i<n; ++i) wv[i]=x[y[i]];
        for(int i=0; i<m; ++i) ws[i]=0;
        for(int i=0; i<n; ++i) ++ws[wv[i]];
        for(int i=1; i<m; ++i) ws[i]+=ws[i-1];
        for(int i=n-1; i>=0; --i) sa[--ws[wv[i]]]=y[i];
        swap(x,y); x[sa[0]]=0; p=1;
        for(int i=1; i<n; ++i) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }

    for(int i=1; i<n; ++i) rank[sa[i]]=i;
    int k=0;
    for(int i=0; i<n-1; height[rank[i++]]=k){
        if(k) --k;
        for(int j=sa[rank[i]-1]; r[i+k]==r[j+k]; ++k);
    }
}

int n,m,r[MAXN],belong[MAXN];
bool isok(int len){
    int cnt=0;
    bool vis[111]={0};
    for(int i=2; i<=n; ++i){
        if(height[i]>=len){
            if(!vis[belong[sa[i]]]){
                vis[belong[sa[i]]]=1;
                ++cnt;
            }
            if(!vis[belong[sa[i-1]]]){
                vis[belong[sa[i-1]]]=1;
                ++cnt;
            }
        }else{
            if(cnt>(m>>1)) return 1;
            memset(vis,0,sizeof(vis));
            cnt=0;
        }
    }
    return 0;
}
void pnt(int len){
    int cnt=0,idx;
    bool vis[111]={0};
    for(int i=2; i<=n; ++i){
        if(height[i]>=len){
            idx=sa[i];
            if(!vis[belong[sa[i]]]){
                vis[belong[sa[i]]]=1;
                ++cnt;
            }
            if(!vis[belong[sa[i-1]]]){
                vis[belong[sa[i-1]]]=1;
                ++cnt;
            }
        }else{
            if(cnt>(m>>1)){
                for(int j=0; j<len; ++j){
                    putchar(r[idx+j]+'a'-1);
                }
                putchar('
');
            }
            memset(vis,0,sizeof(vis));
            cnt=0;
        }
    }
}
int main(){
    char s[1111];
    while(~scanf("%d",&m) && m){
        n=0;
        for(int i=0; i<m; ++i){
            scanf("%s",s);
            for(int j=0; s[j]; ++j){
                belong[n]=i;
                r[n++]=s[j]-'a'+1;
            }
            r[n++]=27+i;
        }
        r[--n]=0;
        SA(r,n+1,127);
        int l=0,r=1000;
        while(l<r){
            int mid=l+r+1>>1;
            if(isok(mid)) l=mid;
            else r=mid-1;
        }
        if(l==0) puts("?");
        else pnt(l);
        putchar('
');
    }
    return 0;
}