这是一道模板题。
读入一个长度为 nn 的由大小写英文字母或数字组成的字符串,请把这个字符串的所有非空后缀按字典序从小到大排序,然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为 11 到 nn 。
输入格式:
一行一个长度为 nn 的仅包含大小写英文字母或数字的字符串。
输出格式:
一行,共n个整数,表示答案。
输入样例#1
ababa
输出样例#1:
5 3 1 4 2
n <= 10^6n<=106
更新了一下自己写的后缀数组
看了一晚上还是有些地方不理解。
明天继续搞
// luogu-judger-enable-o2
#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
const int MAXN=1e6+10;
int sa[MAXN],tax[MAXN],rak[MAXN],tp[MAXN],a[MAXN],N,M;
char s[MAXN];
int pd(int *f,int x,int y,int l)
{
return (f[x]==f[y]&&f[x+l]==f[y+l]);
}
void Q_sort()
{
for(int i=0;i<=M;i++) tax[i]=0;
for(int i=1;i<=N;i++) tax[rak[i]]++;
for(int i=1;i<=M;i++) tax[i]+=tax[i-1];
for(int i=N;i>=1;i--) sa[ tax[rak[tp[i]]]-- ] = tp[i];
}
void S_sort()
{
M=127;
for(int i=1;i<=N;i++) rak[i]=a[i],tp[i]=i;Q_sort();
for(int w=1,p=1;p<N;w=w+w,M=p)
{
p=0;
for(int i=1;i<=w;i++) tp[++p]=N-w+i;
for(int i=1;i<=N;i++)
if(sa[i]>w) tp[++p]=sa[i]-w;
Q_sort();swap(tp,rak);rak[sa[1]]=p=1;
for(int i=2;i<=N;i++)
rak[sa[i]]=pd(tp,sa[i],sa[i-1],w)?p:++p;
}
for(int i=1;i<=N;i++)
printf("%d ",sa[i]);
}
int main()
{
#ifdef WIN32
freopen("a.in","r",stdin);
#else
#endif
scanf("%s",s);
N=strlen(s);
for(int i=0;i<N;i++) a[i+1]=s[i]-48;
S_sort();
}