(http://www.elijahqi.win/2017/07/14/%E3%80%90luogu3809%E3%80%91/%20%E2%80%8E)
题目背景
这是一道模版题。
题目描述
读入一个长度为 的由大小写英文字母或数字组成的字符串,请把这个字符串的所有非空后缀按字典序从小到大排序,然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为 到 。
输入输出格式
输入格式:
一行一个长度为 的仅包含大小写英文字母或数字的字符串。
输出格式:
一行,共n个整数,表示答案。
输入输出样例
输入样例#1:
ababa
输出样例#1:
5 3 1 4 2
说明
敲一敲模板练练手
#include<cstdio>
#include<cstring>
#define N 1100000
char a[N];
int n,k,sa[N],rank[N<<1],rank1[N],count[N],tmp[N],st[255];
int main(){
freopen("3809.in","r",stdin);
freopen("3809.out","w",stdout);
scanf("%s",a+1);
int n=strlen(a+1);
memset(rank,0,sizeof(rank));
memset(st,0,sizeof(st));
memset(rank1,0,sizeof(rank1));
for (int i=1;i<=n;++i)st[a[i]]=1;
for (int i=1;i<=255;++i) st[i]+=st[i-1];
for (int i=1;i<=n;++i) rank[i]=st[a[i]];
k=0;
for (int p=1;k!=n;p<<=1){
memset(count,0,sizeof(count));
for (int i=1;i<=n;++i) count[rank[i+p]]++;
for (int i=1;i<=n;++i) count[i]+=count[i-1];
for (int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
memset(count,0,sizeof(count));
for (int i=1;i<=n;++i) count[rank[i]]++;
for (int i=1;i<=n;++i) count[i]+=count[i-1];
for (int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
memcpy(rank1,rank,sizeof(rank)>>1);
rank[sa[1]]=k=1;
for (int i=2;i<=n;++i){
if (rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
rank[sa[i]]=k;
}
}
for (int i=1;i<=n;++i) printf("%d ",sa[i]);
return 0;
}