传送门

后缀排序模板题。


终于会后缀数组了(然而只会倍增并不会DC3DC3DC3)。

在这里列举几个数组的意思:

sai:sa_i:sai​:当前排名第iii的后缀的起始下标。

rkirk_irki​当前下标为iii的后缀对应的排名。

sa2isa2_isa2i​当前排名为iii的第二关键字对应的下标。

代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
const int N=1e6+5;
int n,m,sa[N],sa2[N],rk[N];
char s[N];
inline void Sort(){
	static int cnt[N];
	for(ri i=1;i<=m;++i)cnt[i]=0;
	for(ri i=1;i<=n;++i)++cnt[rk[i]];
	for(ri i=2;i<=m;++i)cnt[i]+=cnt[i-1];
	for(ri i=n;i;--i)sa[cnt[rk[sa2[i]]]--]=sa2[i];
}
inline void getsa(){
	for(ri i=1;i<=n;++i)rk[i]=s[i]-'0'+1,sa2[i]=i;
	m=127,Sort();
	for(ri w=1,p=0;m!=n;p=0,w<<=1){
		for(ri i=n-w+1;i<=n;++i)sa2[++p]=i;
		for(ri i=1;i<=n;++i)if(sa[i]>w)sa2[++p]=sa[i]-w;
		Sort(),swap(rk,sa2);
		rk[sa[1]]=p=1;
		for(ri i=2;i<=n;++i)rk[sa[i]]=(sa2[sa[i]]==sa2[sa[i-1]]&&sa2[sa[i]+w]==sa2[sa[i-1]+w])?p:++p;
		m=p;
	}
}
int main(){
	scanf("%s",s+1),n=strlen(s+1),getsa();
	for(ri i=1;i<=n;++i)printf("%d ",sa[i]);
	return 0;
}
05-06 06:11