bzoj 2251

 

第一道后缀数组

 

后缀数组要维护三个数组:sa(suffix array), rk(rank)和ht(height)。

含义分别是：

sa[i]:将后缀按照字典序排序后，第i大的后缀的起始位置。

rk[i]:起始位置为i的后缀的排名。

ht[i]:起始位置为i的后缀与排名为rk[i]-1的后缀的最长公共前缀。

 

对于任意一个字串，一定是某个后缀的前缀。

然后从sa[1]开始，统计字串，每个后缀sa[i]的可能的字串的个数（不与前面统计的重复）是：n-sa[i]+1-ht[sa[i]]

然后就这羊枚举子串，幷向后暴力统计个数，这样还能保证字典序是从小到大。

 

/**************************************************************
    Problem: 2251
    User: idy002
    Language: C++
    Result: Accepted
    Time:240 ms
    Memory:892 kb
****************************************************************/
 
#include <cstdio>
#define maxn 3010
 
int n;
int aa[maxn];
int sa[2][maxn], rk[2][maxn], ht[maxn], vv[maxn], p;
char str[maxn];
 
void expand( int k, int sa[maxn], int rk[maxn], int tsa[maxn], int trk[maxn] ) {
    for( int i=1; i<=n; i++ ) vv[rk[sa[i]]]=i;
    for( int i=n; i>=1; i-- ) if( sa[i]>k ) tsa[vv[rk[sa[i]-k]]--]=sa[i]-k;
    for( int i=n-k+1; i<=n; i++ ) tsa[vv[rk[i]]--]=i;
    for( int i=1; i<=n; i++ ) trk[tsa[i]]=trk[tsa[i-1]]+(rk[tsa[i]]!=rk[tsa[i-1]]||rk[tsa[i]+k]!=rk[tsa[i-1]+k]);
}
void makeht() {
    int k=0;
    ht[sa[p][1]] = 0;
    for( int i=1; i<=n; i++ ) {
        if( rk[p][i]==1 ) continue;
        int j=sa[p][rk[p][i]-1];
        while( aa[i+k]==aa[j+k] ) k++;
        ht[i] = k;
        if( k>0 ) k--;
    }
}
void suffix() {
    p=0;
    for( int i=1; i<=n; i++ ) vv[aa[i]]++;
    for( int i=1; i<=2; i++ ) vv[i]+=vv[i-1];
    for( int i=n; i>=1; i-- ) sa[p][vv[aa[i]]--]=i;
    for( int i=1; i<=n; i++ ) rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(aa[sa[p][i]]!=aa[sa[p][i-1]]);
    for( int k=1; k<n; k<<=1,p=1-p ) expand( k, sa[p], rk[p], sa[1-p], rk[1-p] );
    makeht();
}
 
int main() {
    scanf( "%d", &n );
    scanf( "%s", str+1 );
    for( int i=1; i<=n; i++ )
        aa[i] = str[i]-'0'+1;
    suffix();
    for( int i=1; i<=n; i++ ) {
        for( int t=sa[p][i]+ht[sa[p][i]]; t<=n; t++ ) {
            int cnt=1;
            for( int j=i+1; j<=n && ht[sa[p][j]]>=(t-sa[p][i]+1); j++,cnt++);
            if( cnt>1 )
                printf( "%d\n", cnt );
        }
    }
}