bzoj 2251(后缀数组)

2251: [2010Beijing Wc]外星联络

Time Limit: 30 Sec Memory Limit: 256 MB

Submit: 660 Solved: 388

[Submit][Status][Discuss]

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

解题思路：后缀数组然后暴力。细节有点恶心

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<iostream>

using namespace std;

int n,k;

char s[3001];

int sa[3001],rank[3001],height[3001],zan[3001],dis[3001];

inline int read()

{

char y; int x=0,f=1; y=getchar();

while (y<'0' ||y>'9') {if (y=='-') f=-1; y=getchar();}

while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}

return x*f;

}

bool cmp(int x,int y)

{

if (rank[x]<rank[y])return true;

if (rank[x]==rank[y])

{

int u1,u2;

if (x+k<=n) u1=rank[x+k];else u1=-1;

if (y+k<=n) u2=rank[y+k];else u2=-1;

return u1<u2;

}

return false;

}

void makeheight()

{

int j=0;

for (int i=1;i<=n;++i)

{

if (j)--j;

int u1=sa[rank[i]+1];

if (u1==0)continue;

while (i+j<=n && u1+j<=n && s[i+j-1]==s[u1+j-1]) ++j;

height[rank[i]]=j;

}

}

int main()

{

n=read();

scanf("%s",s);

for (int i=1;i<=n;++i)

{

sa[i]=i; rank[i]=int(s[i-1]);

}

for (k=1;k<=n;k=k*2)

{

sort(sa+1,sa+n+1,cmp);

zan[sa[1]]=1;

for (int i=2;i<=n;++i)

{

if (cmp(sa[i-1],sa[i])) zan[sa[i]]=zan[sa[i-1]]+1;else

zan[sa[i]]=zan[sa[i-1]];

}

for (int i=1;i<=n;++i)

rank[i]=zan[i];

}

makeheight();

memset(zan,0,sizeof(zan));

for (int i=1;i<=n-1;++i)

{

memset(dis,0,sizeof(dis));

memset(sa,0,sizeof(sa));

int mx=height[i]; dis[mx]=1; ++sa[mx];

for (int j=i+1;j<=n-1;++j)

{

mx=min(mx,height[j]);

++dis[mx]; ++sa[mx];

}

for (int j=n-1;j>=1;--j)

{

zan[j]+=zan[j+1];

dis[j]+=dis[j+1]-zan[j];

}

for (int j=1;j<=n;++j)

{

if (dis[j]+1>1)printf("%d\n",dis[j]+1);

zan[j]=sa[j];

}

}

}