TYVJ1860 后缀数组

就是模板而已。

然而最后两个点倍增TLE，听说非要DC3算法才能过？

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int mxn=300010;
int sa[mxn],ht[mxn];
int wa[mxn],wb[mxn],wv[mxn];
int cnt[mxn],rk[mxn];
bool cmp(int *r,int a,int b,int k){
    return (r[a]==r[b] && r[a+k]==r[b+k]);
}
void SA(int *r,int n,int m){
    int i,j;
    for(i=0;i<m;i++)cnt[i]=0;
    for(i=0;i<n;i++)cnt[wa[i]=r[i]]++;
    for(i=1;i<m;i++)cnt[i]+=cnt[i-1];
    for(i=n-1;i>=0;i--)sa[--cnt[wa[i]]]=i;
    int p=1;
    for(j=1;p<n;j<<=1,m=p){
        for(p=0,i=n-j;i<n;i++)wb[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)wb[p++]=sa[i]-j;
        for(i=0;i<n;i++)wv[i]=wa[wb[i]];
        for(i=0;i<m;i++)cnt[i]=0;
        for(i=0;i<n;i++)++cnt[wv[i]];
        for(i=1;i<m;i++)cnt[i]+=cnt[i-1];
        for(i=n-1;i>=0;i--)sa[--cnt[wv[i]]]=wb[i];
        swap(wa,wb);
        wa[sa[0]]=0;
        for(p=1,i=1;i<n;i++){
            wa[sa[i]]=cmp(wb,sa[i-1],sa[i],j)?p-1:p++;
        }
    }
    return;
}
void cal_height(int *s,int n){
    int i,j,k=0;
    for(i=1;i<=n;i++)rk[sa[i]]=i;
    for(i=0;i<n;i++){
        if(k)k--;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k])k++;
        ht[rk[i]]=k;
    }
    return;
}
char s[mxn];
int r[mxn];
int main(){
    int i,j;
    scanf("%s",&s);
    int len=strlen(s);
    for(i=0;i<len;i++)r[i]=s[i]-'a'+1;
    r[len]=0;
    SA(r,len+1,30);
    for(i=1;i<=len;i++)printf("%d ",sa[i]+1);
    printf("\n");
    cal_height(r,len);
    for(i=1;i<=len;i++)printf("%d ",ht[i]);
    printf("\n");
    return 0;
}