luogu P4051 [JSOI2007]字符加密

Posted 2021-01-28 lovedsr

前言

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

思路

先将子串复制一遍，组成长度为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;
}