BZOJ2251[2010Beijing Wc]外星联络

BZOJ2251[2010Beijing Wc]外星联络

题目描述

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

输入

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

输出

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

---

Solution

既然看到对应子串的字典序,就很容易想到后缀数组
具体实现的方法其实就是暴力毕竟$n^2$的数据
先$nlogn$求出$sa和height$数组
然后暴力扫一遍就好

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 30010
int t1[maxn],t2[maxn],c[maxn];
int sa[maxn],rak[maxn],height[maxn];
char str[maxn],s[maxn];
void SA(int n){
    int *x=t1,*y=t2,m=10000;
    for(int i=1;i<=m;++i) c[i]=0;
    for(int i=1;i<=n;++i) c[x[i]=s[i]]++;
    for(int i=1;i<=m;++i) c[i]+=c[i-1];
    for(int i=n;i;--i) sa[c[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1){
        int p=0;
        for(int i=n-k+1;i<=n;++i) y[++p]=i;
        for(int i=1;i<=n;++i) if(sa[i]>k) y[++p]=sa[i]-k;
        for(int i=1;i<=m;++i) c[i]=0;
        for(int i=1;i<=n;++i) c[x[y[i]]]++;
        for(int i=1;i<=m;++i) c[i]+=c[i-1];
        for(int i=n;i;--i) sa[c[x[y[i]]]--]=y[i];
        swap(x,y),x[sa[1]]=1,p=1;
        for(int i=2;i<=n;++i) 
        x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p;
        if(p>=n) break;
        m=p;
    }
    for(int i=1;i<=n;++i) rak[sa[i]]=i;
    int k=0;
    for(int i=1;i<=n;++i){
        if(k) --k;
        int j=sa[rak[i]-1];
        while(s[i+k]==s[j+k]) ++k;
        height[rak[i]]=k;
    }
    return ;
}
int main(){
    int n;
    scanf("%d%s",&n,str+1); 
    for(int i=1;i<=n;++i) s[i]=str[i]-'0'+1;
    SA(n);
    int maxx=0;
    height[n+1]=0;//调试的时候发现sa和height鬼畜地被随机赋值了,就懒的用memset
    for(int i=2;i<=n;++i){
        if(height[i]>height[i-1]){
            for(int j=maxx+1;j<=height[i];++j){
                int k=i,now=1;
                while(height[k]>=j){
                    ++now,++k;
                }
                printf("%d\n",now);
            }
        }
        maxx=height[i];
    }
    return 0;
}