【BZOJ2251】外星联络 后缀数组

题目大意是找出一个字符串中所有出现次数大于1的字串并且按字典序输出出现次数

height数组上乱搞。。。

为了防止统计重复,我们统计时可以统计从sa[i]开始的字串,也就是长度Len>=height[i]+1的字串,这样可以保证不重复。

/**************************************************************
    Problem: 2251
    User: cqyzhb
    Language: C++
    Result: Accepted
    Time:120 ms
    Memory:1444 kb
****************************************************************/
 
#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define MAXN 6005
int N,A[MAXN],M;
char s[MAXN];
void Init(){    scanf("%d",&N); scanf("%s",s+1); for(int i=1;i<=N;i++)A[i]=s[i]-'0'+1; M=2;return ;  }
int wa[MAXN],wb[MAXN],wc[MAXN],sa[MAXN],h[MAXN],rank[MAXN];
void Debug()
{
    for(int i=1;i<=N;printf("%d ",A[i]),i++);
    putchar('\n');
    for(int i=1;i<=N;printf("%d ",sa[i]),i++);
}
void make_sa()
{
    int *t,*x=wa,*y=wb,i,k,cnt;
    for(i=1;i<=M;i++)wc[i]=0;
    for(i=1;i<=N;i++)wc[x[i]=A[i]]++;
    for(i=1;i<=M;i++)wc[i]+=wc[i-1];
    for(i=N;i>=1;i--)sa[wc[x[i]]--]=i;
    for(k=1;k<=N;k=k*2)
    {
        cnt=0;
        for(i=N-k+1;i<=N;i++)y[++cnt]=i;
        for(i=1;i<=N;i++)if(sa[i]>k)y[++cnt]=sa[i]-k;
        for(i=1;i<=M;i++)wc[i]=0;
        for(i=1;i<=N;i++)wc[x[y[i]]]++;
        for(i=1;i<=M;i++)wc[i]+=wc[i-1];
        for(i=N;i>=1;i--)sa[wc[x[y[i]]]--]=y[i];
        for(t=x,x=y,y=t,cnt=1,x[sa[1]]=1,i=2;i<=N;i++)
        x[sa[i]]= (y[sa[i-1]]==y[sa[i]])&&(y[sa[i-1]+k]==y[sa[i]+k]) ? cnt : ++cnt ;
        M=cnt; if(cnt==N)break;
    }
    for(i=1;i<=N;i++)rank[sa[i]]=i;
    k=0;
//  Debug();
    for(i=1;i<=N;h[rank[i]]=k,i++)
    {
        if(k)k--;
        for(int j=sa[rank[i]-1];A[i+k]==A[j+k];k++);
    }
    return ;
}
int main()
{
    //freopen("in.txt","r",stdin);
    Init();
    make_sa();
    for(int i=1;i<=N;i++)
    {
        for(int j=h[i]+1;sa[i]+j-1<=N;j++)
        {
            int l=i,r;
        //  for(l=i;h[l]>=j && l>=1;l--);
            for(r=i+1;h[r]>=j && r<=N;r++);
            if(r-l>1)printf("%d\n",r-l);
        }
    }
    return 0;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_34637390/article/details/51558549
个人分类: 字符串 后缀数组
上一篇【BZOJ4104】【Thusc2015】解密运算
下一篇【BZOJ1717】产奶的模式 后缀数组
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭