洛谷P3809（后缀数组模板题）

P3809

 

思路：

使用后缀数组以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;
}