bzoj 2251 后缀数组

建出后缀数组，按字典序深搜一遍整个后缀数组。
记录当前深度和左右端点，对于当前层找出左右端点的01分界点，然后分别搜两边。

#include <bits/stdc++.h>
using namespace std;
#define N 3100
int n;
char s[N];
int sa[N],rank[N],tr[N],has[N];
int cmp(int x,int y,int k)
{
    if(x+k>n||y+k>n)return 0;
    return rank[x]==rank[y]&&rank[x+k]==rank[y+k];
}
void getsa()
{
    int i,cnt;
    for(i=1;i<=n;i++)has[s[i]-'0'+1]++;
    for(i=1,cnt=0;i<=2;i++)if(has[i])tr[i]=++cnt;
    for(i=1;i<=n;i++)has[i]+=has[i-1];
    for(i=1;i<=n;i++)rank[i]=tr[s[i]-'0'+1],sa[has[s[i]-'0'+1]--]=i;
    for(int k=1;cnt!=n;k<<=1)
    {
        for(i=1;i<=n;i++)has[i]=0;
        for(i=1;i<=n;i++)has[rank[i]]++;
        for(i=1;i<=n;i++)has[i]+=has[i-1];
        for(i=n;i>=1;i--)if(sa[i]>k)tr[sa[i]-k]=has[rank[sa[i]-k]]--;
        for(i=1;i<=k;i++)tr[n-i+1]=has[rank[n-i+1]]--;
        for(i=1;i<=n;i++)sa[tr[i]]=i;
        for(i=1,cnt=0;i<=n;i++)tr[sa[i]]=cmp(sa[i],sa[i-1],k) ? cnt:++cnt;
        for(i=1;i<=n;i++)rank[i]=tr[i];
    }
}
void dfs(int l,int r,int d)
{
    int i;
    if(d&&l<r)printf("%d\n",r-l+1);
    for(;sa[l]+d>n&&l<=r;l++);
    if(l>r)return;
    for(i=l;i<=r&&s[sa[i]+d]=='0';i++);i--;
    dfs(l,i,d+1);
    dfs(i+1,r,d+1);
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d",&n);
    scanf("%s",s+1);
    getsa();
    dfs(1,n,0);
    return 0;
}