后缀数组 模板

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

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

留着以后用....

#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)。