后缀数组(倍增) 时间nlogn 空间n

模板   

struct SuffixArray{
    int sa[maxn];//排名为i的后缀的起始下标
    int Rank[maxn];//起始下标为i的后缀排名
    int height[maxn];//排名为i和i-1的最长公共前缀
    int wa[maxn],wb[maxn],wv[maxn],ws[maxn];

    int cmp(int *r, int a,int b,int l){//双关键字排序的比较
        return r[a]==r[b] && r[a+l]==r[b+l];
    }

    void da(int *r,int n,int m){
        ///r表示原数组,sa表示排名为i的下标,n表示r的大小+1,m表示字符的范围
        ///注意使用时如果r的大小为n,那么应传入n+1
        //x为第一关键字，y为第二关键字
        int i,k,p,*x = wa,*y = wb,*t;
        //长度为1时的基数排序
        for(i=0;i<m;i++) ws[i] = 0;
        for(i=0;i<n;i++) ws[x[i] = r[i]] ++;
        for(i=0;i<m;i++) ws[i] += ws[i-1];
        for(i=n-1;i>=0;i--) sa[-- ws[x[i]]] = i;
        //主要循环,k倍增
        for(k=1,p=1; p<n; k*=2, m=p){
            //以下两行代码实现了对第二关键字的排序
            for(p=0,i=n-k;i<n;i++) y[p++] = i;
            for(i=0;i<n;i++){
                if(sa[i] >= k){
                    y[p++] = sa[i] - k;
                }
            }
            //提取出每个子串的第一关键字
            for(i=0;i<n;i++){
                wv[i] = x[y[i]];
            }
            //基数排序
            for(i=0;i<m;i++) ws[i]=0;
            for(i=0;i<n;i++) ws[wv[i]]++;
            for(i=1;i<m;i++) ws[i]+=ws[i-1];
            for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
            t=x,x=y,y=t;
            for(p=1,x[sa[0]]=0,i=1;i<n;i++)
                x[sa[i]]=cmp(y,sa[i-1],sa[i],k)?p-1:p++;
        }
    }

    //求rank数组与height数组
    void cal(int *r,int n){
        ///此处传入n即可
        int i,j,k=0;
        for(i=1;i<=n;i++) Rank[sa[i]] = i;
        for(i=0;i<n;i++){
            if(k > 0) k--;
            j=sa[Rank[i]-1];
            while(r[i+k] == r[j+k]) k++;
            height[Rank[i]] = k;
        }

        //for(i=0;i<n;height[Rank[i++]]=k)
        //for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);
    }

    /*使用方法
    da(r,n+1,m);
    cal(r,n);*/
}SA;