poj3294 Life Forms（后缀数组）

 

【题目链接】

 

　　http://poj.org/problem?id=3294　

 

【题意】

 

　　多个字符串求出现超过R次的最长公共子串。

 

【思路】

    

       二分+划分height，判定一个组中是否包含不小于R个不同字符串的后缀。

       需要注意的有：

　　　　　1)      c[]尽量开大，字符范围为“偏移”之后的范围。

　　　　　2)      用kase作为标记节省了每次开始新段需要清零的时间。

　　　　　3)      因为height是sa[i]与sa[i-1]的关系，所以无论是在can的开始还是在新段开始都需要初始为一个串的情况。

【代码】

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

const int maxn = 200000+10;

int s[maxn];
int sa[maxn],c[maxn],t[maxn],t2[maxn];

void build_sa(int m,int n) {
    int i,*x=t,*y=t2;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[i]]++;
    for(i=1;i<m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    
    for(int k=1;k<=n;k<<=1) {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=0;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        
        swap(x,y);
        p=1; x[sa[0]]=0;
        for(i=1;i<n;i++) 
            x[sa[i]]=y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
int rank[maxn],height[maxn];
void getHeight(int n) {
    int i,j,k=0;
    for(i=0;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;i++) {
        if(k) k--;
        j=sa[rank[i]-1];
        while(s[j+k]==s[i+k]) k++;
        height[rank[i]]=k;
    }
}

int T;
char a[maxn];

int f[200],kase;
vector<int> st;
int can(int limit,int n,int len) {
    int cnt=1,ok=0;
    st.clear();
    f[sa[1]/len]=kase;
    for(int i=2;i<=n;i++) {
        if(height[i]<limit) {
            cnt=1;
            f[sa[i]/len]=++kase;               //检查每一个组中 
        }
        else {
            if(f[sa[i]/len]!=kase) {
                f[sa[i]/len]=kase;
                if(cnt>=0) cnt++;
                if(cnt>T/2) {
                    ok=1;
                    st.push_back(sa[i]);
                    cnt=-1;
                }
            }
        }
    }
    return ok;
}
void init() {
    kase=1;
    memset(sa,0,sizeof(sa));
    memset(f,0,sizeof(f));
}
int main() {
    //freopen("in.in","r",stdin);
    //freopen("out.out","w",stdout);
    while(scanf("%d",&T)==1 && T) {
        init();
        int len,n=0;
        for(int i=0;i<T;i++) {
            scanf("%s",&a);
            len=strlen(a);
            for(int j=0;j<len;j++) s[n++]=a[j]+100;
            s[n++]=i+1;
        }
        n--;
        s[n]=0;
        
        build_sa(250,n+1);
        getHeight(n);
        
        int L=0,R=len+1;
        while(L<R) {
            int M=L+(R-L+1)/2;
            if(can(M,n,len+1)) L=M;
            else R=M-1;
        }
        can(L,n,len+1);                        //再调用一次求出st 
        if(L==0) printf("?
");
        else {
            for(int i=0;i<st.size();i++)  {
                for(int j=st[i];(j-st[i]+1)<=L;j++)
                    printf("%c",s[j]-100);
                putchar('
');
            }
        }
        putchar('
');
    }
    return 0;
}