后缀数组 模板

搞了一下午才把后缀数组稍微明白了点  真是弱呀~~

根据网上各种大神的模板  找了一份自己好看的

留着以后用....

#include <stdio.h>#include <cstring>const int maxn =200010;char s[maxn];int n;int st[maxn],sa[maxn],rank[maxn],height[maxn],wa[maxn],wb[maxn],ws[maxn],wv[maxn];int cmp(int *r,int a,int b,int l){    return r[a]==r[b] && r[a+l]==r[b+l];}int da(char *r,int n,int m){    int i,j,p,*x = wa,*y = wb,*t;    for (i = 0; i < m; ++i) ws[i] = 0;    for (i = 0; i < n; ++i) ws[ x[i] = r[i]  ]++;    for (i = 1; i < m; ++i) ws[i] += ws[i-1];    for (i = n-1; i >= 0; --i) sa[--ws[x[i]]] = i;    for (j = 1,p = 1; p < n; j *= 2,m = p)    {        for (p = 0,i = n-j; i < n; ++i) y[p++] = i;        for (i = 0; i < n; ++i) if (sa[i] >= j) y[p++] = sa[i] - j;        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];        for (t = x,x = y,y = t,p = 1,x[sa[0]] = 0,i = 1; i < n; ++i)            x[sa[i]] = cmp(y,sa[i-1],sa[i],j)? p-1:p++;    }    return 0;}int calheight(char *r,int *sa,int n){    int i,j,k = 0;    for (i = 1; i <= n; ++i) rank[sa[i]] = i;    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++);    return 0;}int main(){    while(~scanf("%s",s))    {        n=strlen(s);        int m=400;        da(s,n+1,m);        calheight(s,sa,n);        for(int i=1; i<=n; i++)  //都是从下标1开始的            printf("%d ",sa[i]);        printf("\n");        for(int i=1; i<=n; i++)            printf("%d ",height[i]);        printf("\n");    }    return 0;}//以上是模板,调用da(）时应这样， da(r,sa,n+1,m+1)；//calheight则常规调用即可 da(r,sa,n)。