HDU3518 后缀数组求不可重叠重复出现的不同子串个数

枚举子串长度，根据height分组，如果本组sa最小值与sa最大值之差超过枚举的长度，则本组对于答案贡献为1。

 #include <iostream>
 #include <vector>
 #include <algorithm>
 #include <string>
 #include <string.h>
 #include <stdio.h>
 #include <queue>
 #include <stack>
 #include <map>
 #include <set>
 #include <cmath>
 #include <ctime>
 #include <cassert>
 #include <sstream>
 using namespace std;

 const int N=;
 const int MOD=1e9+;

 char s[N];
 struct SuffixArray {;
     int sa[N];
     int t1[N],t2[N],c[N];
     int rk[N],height[N];

     inline int cmp(int *r,int a,int b,int l){
         return r[a]==r[b]&&r[a+l]==r[b+l];
     }
     void calcSA (char *s,int n,int m) {
         int i,j,p,*x=t1,*y=t2;
         for(i=;i<m;i++)c[i]=;
         for(i=;i<n;i++)c[x[i]=s[i]]++;
         for(i=;i<m;i++)c[i]+=c[i-];
         for(i=n-;i>=;i--)sa[--c[x[i]]]=i;
         for(j=;j<=n;j<<=){
             p=;
             for(i=n-j;i<n;i++)y[p++]=i;
             for(i=;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j; // 排名从小到大，如果pos比j大，则suffix(sa[i]-j)的第二关键字为p
             for(i=;i<m;i++)c[i]=;
             for(i=;i<n;i++)c[x[y[i]]]++;
             for(i=;i<m;i++)c[i]+=c[i-];
             for(i=n-;i>=;i--)sa[--c[x[y[i]]]]=y[i]; // 根据第二关键字从大到小，确定新一轮sa
             swap(x,y);
             p=;x[sa[]]=;
             for(i=;i<n;i++)
                 x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
             if(p>=n)break;
             m=p;
         }
     }
     void calcHeight(char *s,int n) {
         int i,j,k=;
         for(i=;i<=n;i++)rk[sa[i]]=i;
         for(i=;i<n;i++){
             if(k)k--; // h[i]>=h[i-1]-1
             j=sa[rk[i]-]; // suffix(j)排名在suffix(i)前一位
             while(s[i+k]==s[j+k])k++; // 暴力计算lcp
             height[rk[i]]=k;
         }
     }
     int lcp(int a,int b,int len) {
         if (a==b) return len-a;
         int ra=rk[a],rb=rk[b];
         if (ra>rb) swap(ra,rb);
         return queryST(ra+,rb);
     }

     int st[N][];//第二维要保证大于log(N<<1)
     void initST(int n) {
         for (int i=; i<=n; i++)
             st[i][]=height[i];
         for (int j=; (<<j)<=n; j++) {
             int k=<<(j-);
             for (int i=; i+k<=n; i++)
                 st[i][j]=min(st[i][j-],st[i+k][j-]);
         }
     }
     int queryST(int a,int b) {
         if (a>b) swap(a,b);
         int dis=b-a+;
         int k=log((double)dis)/log(2.0);
         return min(st[a][k],st[b-(<<k)+][k]);
     }
 }suf;

 int main () {
     while (scanf("%s",s)!=EOF){
         if (s[]=='#') break;
         int n=strlen(s);
         suf.calcSA(s,n+,);
         suf.calcHeight(s,n);
         //for (int i=0;i<=n;i++) cout<<suf.sa[i]<<" ";cout<<endl;
         //for (int i=0;i<=n;i++) cout<<suf.height[i]<<" ";cout<<endl;
         int ret=;
         for (int i=;i<=n/;i++) {
             int mi=suf.sa[],ma=suf.sa[];
             for (int j=;j<=n;j++) {
                 if (suf.height[j]<i) {
                     if (mi+i<=ma) ret++;
                     mi=suf.sa[j];
                     ma=suf.sa[j];

                 }
                 mi=min(mi,suf.sa[j]);
                 ma=max(ma,suf.sa[j]);
             }
             if (mi+i<=ma) ret++;
         }
         cout<<ret<<endl;
     }
     return ;
 }