bzoj2251 [2010Beijing WC]外星联络(后缀数组+暴力)

求所有出现次数大于1的字串。。用后缀数组处理出h数组,每个子串第一次出现时,往后扫,看重复了几次。n=3000,暴力即可。

#include 
#include 
#define N 3010
int n,m,rank[N<<1],rank1[N],sa[N],count[N],tmp[N],h[N];
char s[N];
int main(){
//  freopen("a.in","r",stdin);
    scanf("%d%s",&n,s+1);
    for(int i=1;i<=n;++i){
        if(s[i]=='0') rank[i]=1;else rank[i]=2;
    }m=2;int k=0;
    for(int p=1;k!=n;p<<=1,m=k){
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[i+p]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
        memcpy(rank1,rank,sizeof(rank1));
        rank[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
            rank[sa[i]]=k;
        }
    }
    for(int i=1;i<=n;++i){
        if(rank[i]==1){h[1]=0;continue;}
        if(i==1||h[rank[i-1]]<=1) k=0;
        if(k) --k;
        while(s[i+k]==s[sa[rank[i]-1]+k]) ++k;
        h[rank[i]]=k;
    }
    for(int i=1;i<=n;++i)
        for(int j=h[i]+1;sa[i]+j-1<=n;++j){//第一次出现s[sa[i]...sa[i]+j-1]这个子串 
            int r=0;for(r=i+1;r<=n&&h[r]>=j;++r);//看后面还有几个这样的子串 
            if(r-i>1) printf("%d\n",r-i);
        }
    return 0;
}

你可能感兴趣的:(SA,bzoj)