luogu P4051 [JSOI2007]字符加密

前言

其实就是个后缀数组模板题
可还是有几个的地方不太明白

思路

先将子串复制一遍，组成长度为2*n的子串
给出的子串一定会在前n个后缀
而且后面的优先级不会影响前面的相对大小
然后求得sa输出就好
输出的时候把没有必要输出的忽略掉就好

代码

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
#define ROF(i,a,b) for(int i=a;i>=b;--i)
using namespace std;
const int maxn=2e5+7;
char s[maxn];
int n,m,sa[maxn],rk[maxn],x[maxn],c[maxn];
void get_sa() {
	FOR(i,1,n) ++c[rk[i]=s[i]];
	FOR(i,2,m) c[i]+=c[i-1];
	ROF(i,n,1) sa[c[rk[i]]--]=i;
	for(int k=1;k<=n;k<<=1) {
		int p=0;
		FOR(i,n-k+1,n) x[++p]=i;
		FOR(i,1,n) if(sa[i]>k) x[++p]=sa[i]-k;
		FOR(i,1,m) c[i]=0;
		FOR(i,1,n) ++c[rk[i]]; 
		FOR(i,2,m) c[i]+=c[i-1];
		ROF(i,n,1) sa[c[rk[x[i]]]--]=x[i],x[i]=0;
		swap(rk,x),rk[sa[1]]=1,p=1;
		FOR(i,2,n) rk[sa[i]]=(x[sa[i]]==x[sa[i-1]]&&x[sa[i]+k]==x[sa[i-1]+k])?p:++p;
		if(p==n) break;m=p;
	}
}
int main() {
	scanf("%s",s+1);
	n=strlen(s+1);
	FOR(i,1,n) s[n+i]=s[i];
	n<<=1,m=150;
	get_sa();
	FOR(i,1,n) {
		if(sa[i]>n/2) continue;
		else cout<<s[sa[i]+n/2-1];
	}
	return 0;
}