后缀数组 模板

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