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[][maxn], rk[][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=; i<=n; i++ ) vv[rk[sa[i]]]=i;
     for( int i=n; i>=; i-- ) if( sa[i]>k ) tsa[vv[rk[sa[i]-k]]--]=sa[i]-k;
     for( int i=n-k+; i<=n; i++ ) tsa[vv[rk[i]]--]=i;
     for( int i=; i<=n; i++ ) trk[tsa[i]]=trk[tsa[i-]]+(rk[tsa[i]]!=rk[tsa[i-]]||rk[tsa[i]+k]!=rk[tsa[i-]+k]);
 }
 void makeht() {
     int k=;
     ht[sa[p][]] = ;
     for( int i=; i<=n; i++ ) {
         if( rk[p][i]== ) continue;
         int j=sa[p][rk[p][i]-];
         while( aa[i+k]==aa[j+k] ) k++;
         ht[i] = k;
         if( k> ) k--;
     }
 }
 void suffix() {
     p=;
     for( int i=; i<=n; i++ ) vv[aa[i]]++;
     for( int i=; i<=; i++ ) vv[i]+=vv[i-];
     for( int i=n; i>=; i-- ) sa[p][vv[aa[i]]--]=i;
     for( int i=; i<=n; i++ ) rk[p][sa[p][i]]=rk[p][sa[p][i-]]+(aa[sa[p][i]]!=aa[sa[p][i-]]);
     for( int k=; k<n; k<<=,p=-p ) expand( k, sa[p], rk[p], sa[-p], rk[-p] );
     makeht();
 }

 int main() {
     scanf( "%d", &n );
     scanf( "%s", str+ );
     for( int i=; i<=n; i++ )
         aa[i] = str[i]-''+;
     suffix();
     for( int i=; i<=n; i++ ) {
         for( int t=sa[p][i]+ht[sa[p][i]]; t<=n; t++ ) {
             int cnt=;
             for( int j=i+; j<=n && ht[sa[p][j]]>=(t-sa[p][i]+); j++,cnt++);
             if( cnt> )
                 printf( "%d\n", cnt );
         }
     }
 }