BZOJ 2251 2010 beijing Wc 外星联络

2251: [2010Beijing Wc]外星联络

Time Limit: 30 Sec  Memory Limit: 256 MB
Submit: 1045  Solved: 644
[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 

 

Source

我们知道一个字符的所有子串都可以用后缀的前缀表示出来   所以我们先用那一套把sa和height搞出来

然后我们考虑本质不同的重复子串，我们从rank[1]枚举到rank[n]，有不同子串当从sa[i]开始至少到sa[i]+h[i]+1结束时

然后判断一下有没有重复的就行了

#include <bits/stdc++.h>
#define ll long long
#define inf 1e9+10
using namespace std;
inline int read(){
	int x=0;int f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
const int MAXN=1e6+10;
int sa[MAXN],h[MAXN],rank[MAXN],n,tmp[MAXN],cnt[MAXN],p[MAXN];
char ch[MAXN];
inline bool equ(int x,int y,int l){
	return rank[x]==rank[y]&&rank[x+l]==rank[y+l]; 
}
void build_suffix(){
	for(int i=1;i<=n;i++) {sa[i]=i;rank[i]=ch[i]-'0'+1;}
	for(int i,pos=0,sig=5,l=0;pos<n;sig=pos){
		for(i=n-l+1,pos=0;i<=n;i++) p[++pos]=i;
		for(i=1;i<=n;i++) if(sa[i]>l) p[++pos]=sa[i]-l;	
		for(i=0;i<=sig;i++) cnt[i]=0;
		for(i=1;i<=n;i++) cnt[rank[i]]++;
		for(i=1;i<=sig;i++) cnt[i]+=cnt[i-1];
		for(i=n;i>=1;i--) sa[cnt[rank[p[i]]]--]=p[i];
		for(i=1,pos=0;i<=n;i++) tmp[sa[i]]=equ(sa[i],sa[i-1],l)?pos:++pos;
		for(i=1;i<=n;i++) rank[i]=tmp[i];
		l=!l?1:l<<1;
	}
}
void build_height(){	
	for(int i=1,k,j=0;i<=n;i++){
		if(!(k=sa[rank[i]-1])) {j=0;continue;}
		if(j) j--;
		while(ch[k+j]==ch[i+j]) j++;
		h[rank[i]]=j;
	}
}
int main(){
	n=read();
	scanf("%s",ch+1);
	build_suffix();
	build_height();
	for(int i=1;i<=n;i++){
		for(int j=h[i]+1;sa[i]+j-1<=n;j++){
			int r;
			for(r=i+1;r<=n&&h[r]>=j;r++);
			if(r-i>=2) printf("%d\n",r-i);
		}
	}
	return 0;
}
/*
7  
1010101
*/