外星联络（bzoj 2251）

Posted 2020-08-28 Cola

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 

/*
  这个题需要动一点脑子。
  先求出sa和height数组，然后对于每个i，枚举一个大于2的长度j，往后扫跟它height>=j的元素。 
  为了避免重复，j每次枚举时从height[i-1]开始枚举。 
*/
#include<cstdio>
#include<iostream>
#define N 3010
using namespace std;
int s[N],t1[N],t2[N],c[N],sa[N],rk[N],ht[N],n,m=100;
bool cmp(int *y,int a,int b,int k){
    int a1=y[a],b1=y[b];
    int a2=a+k<n?y[a+k]:-1;
    int b2=b+k<n?y[b+k]:-1;
    return a1==b1&&a2==b2;
}
void getsa(){
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;~i;i--) sa[--c[x[i]]]=i;
    for(int k=1,p=0;k<=n;k*=2,m=p,p=0){
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;~i;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++)
            if(cmp(y,sa[i-1],sa[i],k)) x[sa[i]]=p-1;
            else x[sa[i]]=p++;
        if(p>=n) break;
    }
}
void getht(){
    for(int i=0;i<n;i++) rk[sa[i]]=i;
    for(int i=0,j,k=0;i<n;ht[rk[i++]]=k){
        if(!rk[i]) continue;
        j=sa[rk[i]-1];k=k?k-1:0;
        while(i+k<n&&j+k<n&&s[i+k]==s[j+k]) k++;
    }ht[0]=0;
}
int main(){
    char ch[N];scanf("%d%s",&n,ch);
    for(int i=0;i<n;i++) s[i]=ch[i];
    getsa();getht();
    int pre=0;
    for(int i=1;i<n;i++){
        for(int j=pre+1;j<=ht[i];j++){
            int ans=2,k=i+1;
            while(ht[k]>=j) k++,ans++;
            printf("%d\n",ans);
        }
        pre=ht[i];
    }
    return 0;
}