思路:
使用后缀数组以O(nlogn)的复杂度将所有后缀数组的字典序大小求出即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 4e6+10;
//ss数组表示将要输入的字符串数组
char ss[maxn];
//n表示字符串ss的长度,
//m表示排名的范围,就是基数排序中所需要的桶的数量,每次基数排序之后都会发生变化
//x表示字符的字典序的排名,一开始是0-127,之后随着m的范围不断缩小
//y数组表示第二关键字按照字典序排序后的后缀字符串的编号
//c数组是基数排序中的桶
//sa表示按照字典序排序,排名为i的后缀数组的首字母的编号是sa[i]
int n,m,x[maxn],y[maxn],c[maxn],sa[maxn];
void build_sa(){
int i,j;
//第一次基数排序
for(i=0;i<m;i++) c[i] = 0;
for(i=0;i<n;i++) c[x[i] = ss[i]]++;
for(i=1;i<m;i++) c[i] += c[i-1];
for(i=n-1;i>=0;i--) sa[--c[x[i]]] = i;
//sa[i]数组表示排名第i个的后缀字符串的标号是sa[i].
//倍增log(n)次比较
for(int k = 1;k<=n;k<<=1){
int p = 0;
//利用sa将第二次关键字排序
for(i=n-k;i<n;i++) y[p++] = i; //后面第二位是0的一定排在前面
for(i=0;i<n;i++)
if(sa[i]>=k) y[p++] = sa[i] - k; //从排好序的sa数组中找剩下的有序的数组序列,而且必须大于k,因为之前的排名少了k个。
for(i=0;i<m;i++) c[i] = 0;
for(i=0;i<n;i++) c[x[y[i]]]++; //表示在第二关键字排好序的基础上对第一关键字进行排序
//基数排序
for(i=1;i<m;i++) c[i] += c[i-1];
for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]] = y[i];
swap(x,y);
//初始化序列的第一个元素,编号为0
x[sa[0]] = 0;p = 1;
for(i=1;i<n;i++)
if(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) x[sa[i]] = p-1;
else x[sa[i]] = p++;
//p>=n表示已经统计了n次,就是每个数都有自己独特的排名,所以继续倍增也不会改变,直接退出即可。
if(p>=n) break;
m = p;
}
for(i=0;i<n;i++){
if(i) printf(" ");
printf("%d",sa[i]+1);
}
printf("\n");
}
int main(void)
{
m = 127;
scanf("%s",ss);
n = strlen(ss);
build_sa();
return 0;
}