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=223456;
const int MOD=1e9+7;


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=0;i<m;i++)c[i]=0;
        for(i=0;i<n;i++)c[x[i]=s[i]]++;
        for(i=1;i<m;i++)c[i]+=c[i-1];
        for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
        for(j=1;j<=n;j<<=1){
            p=0;
            for(i=n-j;i<n;i++)y[p++]=i;
            for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j; // 排名从小到大，如果pos比j大，则suffix(sa[i]-j)的第二关键字为p
            for(i=0;i<m;i++)c[i]=0;
            for(i=0;i<n;i++)c[x[y[i]]]++;
            for(i=1;i<m;i++)c[i]+=c[i-1];
            for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i]; // 根据第二关键字从大到小，确定新一轮sa
            swap(x,y);
            p=1;x[sa[0]]=0;
            for(i=1;i<n;i++)
                x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
            if(p>=n)break;
            m=p;
        }
    }
    void calcHeight(char *s,int n) {
        int i,j,k=0;
        for(i=0;i<=n;i++)rk[sa[i]]=i;
        for(i=0;i<n;i++){
            if(k)k--; // h[i]>=h[i-1]-1
            j=sa[rk[i]-1]; // 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+1,rb);
    }

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

int main () {
    while (scanf("%s",s)!=EOF){
        if (s[0]=='#') break;
        int n=strlen(s);
        suf.calcSA(s,n+1,128);
        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=0;
        for (int i=1;i<=n/2;i++) {
            int mi=suf.sa[1],ma=suf.sa[1];
            for (int j=2;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 0;
}