「SNOI2019」字符串
看起来非常一眼啊,我们完全可以\(std::sort\)来解决这歌问题
于是现在的问题转化成了比较函数怎么写
随便画一下就会发现前面的好几位是一样的,后面的好几位也是一样,只需要比较中间的一段子串的大小就好了
比较大小我们显然是需要\(lcp\)的
发现这里的后缀恰好有一位是错位的,于是我们只需要知道对于每一个后缀\(i\)知道它和后缀\(i+1\)的\(lcp\)长度是多少
我们显然可以\(O(n)\)递推求出来
之后快排即可
#include<bits/stdc++.h>
const int maxn=1e6+5;
int n,het[maxn],a[maxn];
char S[maxn];
inline int cmp(int a,int b) {
int g=(a>b)?b:a,h=(a<b)?b:a;
if(g+het[g]>=h||S[g+het[g]]==S[g+het[g]+1]) return a<b;
if(S[g+het[g]]<S[g+het[g]+1]) return a>b;
return a<b;
}
int main() {
scanf("%d",&n);scanf("%s",S+1);
for(int i=1;i<=n;i++) a[i]=i;
for(int i=n-1;i;--i) het[i]=(S[i]==S[i+1])?het[i+1]+1:0;
std::sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++) printf("%d ",a[i]);
return 0;
}