【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

对于 100%的数据，满足 0 <=  N <=3000

题解
Orz YZH 给YZH神犇跪烂烂

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1000000000
#define mod 1000000007
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*10+ch-'0';ch=getchar();}
    return x*f;
}
char ch[3005];
int n,a[3005],sa[6005],height[6005],rk[6005];
int sta[6005],stb[6005],A[6005],B[6005],tsa[6005];
void SA()
{
    for (int i=1;i<=n;i++) sta[a[i]]++;sta[2]+=sta[1];
    for (int i=n;i;i--) sa[sta[a[i]]--]=i;
    rk[sa[1]]=1;
    for (int i=2;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(a[sa[i]]!=a[sa[i-1]]);
    for (int L=1;rk[sa[n]]!=n;L<<=1)
    {
        for (int i=0;i<=n;i++) sta[i]=stb[i]=0;
        for (int i=1;i<=n;i++) A[i]=rk[i],B[i]=rk[i+L];
        for (int i=1;i<=n;i++) sta[A[i]]++,stb[B[i]]++;
        for (int i=1;i<=n;i++) sta[i]+=sta[i-1],stb[i]+=stb[i-1];
        for (int i=n;i;i--) tsa[stb[B[i]]--]=i;
        for (int i=n;i;i--) sa[sta[A[tsa[i]]]--]=tsa[i];
        rk[sa[1]]=1;
        for (int i=2;i<=n;i++) rk[sa[i]]=rk[sa[i-1]]+(A[sa[i]]!=A[sa[i-1]]||B[sa[i]]!=B[sa[i-1]]);
    }
}
void HEIGHT()
{
    int len=0;
    for (int i=1;i<=n;++i)
    {
        if (len) --len;
        while (a[i+len]==a[sa[rk[i]-1]+len]) ++len;
        height[rk[i]]=len;
    }
}
int main()
{
    n=read();scanf("%s",ch+1);
    for (int i=1;i<=n;i++) a[i]=ch[i]-'0'+1;
    SA();HEIGHT();
    for(int i=1;i<=n;i++)
        for(int j=height[i]+1;sa[i]+j-1<=n;j++)
        {
            int l=i,r;
            for(r=i+1;r<=n&&height[r]>=j;r++);
            if(r-l>1)printf("%d\n",r-l);
        }
    return 0;
}