巴蜀1359 后缀数组

Description 我们定义一个字符串的后缀suffix(i)表示从s[i]到s[length(s)]这段子串。 后缀数组（Suffix

array）SA[i]中存放着一个排列，满足suffix(sa[i])定义height[i]表示suffix(sa[i])与suffix(sa[i-1])之间的最长公共前缀长度，其中height[1]=0

你的任务就是求出SA和height这两个数组。字符串长度<=200000 Input 一行，为描述中的字符串（仅会出现小写字母）

Output 共两行，每行n个数，第一行为sa[i]，第二行为height[i]，其中每行的数均用空格隔开

后缀数组。

发现网上很多模板都是错的，可以试一下’aaaaa’。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
char s[200010];
int sa[200010],cnt[200010],rank[200010],height[200010],t1[200010],t2[200010],n,m;
int main()
{
int i,j,k,p,q,*x=t1,*y=t2;
scanf("%s",s+1);
n=strlen(s+1);
m='z';
for (i=1;i<=n;i++)
cnt[x[i]=s[i]]++;
for (i=2;i<=m;i++)
cnt[i]+=cnt[i-1];
for (i=n;i;i--)
sa[cnt[x[i]]--]=i;
for (k=1;k<=n;k<<=1)
{
p=0;
for (i=n-k+1;i<=n;i++)
y[++p]=i;
for (i=1;i<=n;i++)
if (sa[i]-k>=1)
y[++p]=sa[i]-k;
for (i=1;i<=m;i++)
cnt[i]=0;
for (i=1;i<=n;i++)
cnt[x[y[i]]]++;
for (i=2;i<=m;i++)
cnt[i]+=cnt[i-1];
for (i=n;i;i--)
sa[cnt[x[y[i]]]--]=y[i];
swap(x,y);
p=1;
x[sa[1]]=1;
for (i=2;i<=n;i++)
if (y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k])
x[sa[i]]=p;
else
x[sa[i]]=++p;
if ((m=p)>=n) break;
}
for (i=1;i<=n;i++)
rank[sa[i]]=i;
for (i=1,k=0;i<=n;i++)
{
if (k) k--;
if (rank[i]==1)
{
k=height[1]=0;
continue;
}
j=sa[rank[i]-1];
while (s[i+k]==s[j+k]) k++;
height[rank[i]]=k;
}
for (i=1;i<=n;i++)
printf("%d%c",sa[i],i==n?'\n':' ');
for (i=1;i<=n;i++)
printf("%d%c",height[i],i==n?'\n':' ');
}