【bzoj2251】[2010Beijing Wc]外星联络 后缀数组

Description

小 P 在看过电影《超时空接触》(Contact)之后被深深的打动，决心致力于寻
找外星人的事业。于是，他每天晚上都爬在屋顶上试图用自己的收音机收听外星
人发来的信息。虽然他收听到的仅仅是一些噪声，但是他还是按照这些噪声的高
低电平将接收到的信号改写为由 0 和 1 构成的串， 并坚信外星人的信息就隐藏在
其中。他认为，外星人发来的信息一定会在他接受到的 01 串中重复出现，所以
他希望找到他接受到的 01 串中所有重复出现次数大于 1 的子串。但是他收到的
信号串实在是太长了，于是，他希望你能编一个程序来帮助他。

Input

输入文件的第一行是一个整数N ，代表小 P 接收到的信号串的长度。
输入文件第二行包含一个长度为N 的 01 串，代表小 P 接收到的信号串。

Output

输出文件的每一行包含一个出现次数大于1 的子串所出现的次数。输出的顺
序按对应的子串的字典序排列。

Sample Input

7
1010101

Sample Output

3
3
2
2
4
3
3
2
2

HINT

 

题解：后缀数组，求出height，倍增的方式去求后缀数组

然后就是裸题了。

#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>

#define N 3007
#define inf 2000000007
#define ll long long
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}

int n,p,q,k;
int v[N];
char ch[N];
int a[N],h[N];
int rk[2][N],sa[2][N];

void calsa(int sa[N],int rk[N],int SA[N],int RK[N])
{
    for (int i=1;i<=n;i++)
        v[rk[sa[i]]]=i;
    for (int i=n;i>=1;i--)
        if (sa[i]>k) SA[v[rk[sa[i]-k]]--]=sa[i]-k;
    for (int i=n-k+1;i<=n;i++)
        SA[v[rk[i]]--]=i;
    for (int i=1;i<=n;i++)
        RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
}
void work()
{
    p=0,q=1;
    for (int i=1;i<=n;i++)v[a[i]]++;
    for (int i=1;i<=30;i++)v[i]+=v[i-1];
    for (int i=1;i<=n;i++)sa[p][v[a[i]]--]=i;
    for (int i=1;i<=n;i++)rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i]]!=a[sa[p][i-1]]);
    for (k=1;k<n;k<<=1)
    {
        calsa(sa[p],rk[p],sa[q],rk[q]);
        swap(p,q);
    }
}
void get_height()
{
    k=0;
    for (int i=1;i<=n;i++)
        if (rk[p][i]==1) h[rk[p][i]]=0;
        else
        {
            int j=sa[p][rk[p][i]-1];
            while(a[i+k]==a[j+k])k++;
            h[rk[p][i]]=k;
            if (k>0)k--;
        }
}
int main()
{
    n=read();
    scanf("%s",ch+1);
    for (int i=1;i<=n;i++)
        a[i]=ch[i]-'0'+1;
    work(),get_height();
    for (int i=1;i<=n;i++)
        for (int j=h[i]+1;sa[p][i]+j-1<=n;j++)
        {
            int l,r;
            for (l=i;l>=1&&h[l]>=j;l--);
            for (r=i+1;r<=n&&h[r]>=j;r++);
            if (r-l>1) printf("%d\n",r-l);
        }
}