LG4051/BZOJ1031 「JSOI2007」字符加密 后缀数组

问题描述

BZOJ1031

LG4051

---

题解

发现这是一个环，根据经验，破环为链，于是字符环变为了字符串

之后对这个复制之后的字符串求后缀数组。

$len$代表原字符串长度，代表复制后的字符串长度

最后输出的时候，判断一下，如果$SA_i \le len$，则输出$str_i$。

---

Code

#include<bits/stdc++.h>
using namespace std;

#define maxn 1000007

void read(int &x){
    x=0;char ch=1;int fh;
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') fh=-1,ch=getchar();
    else fh=1;
    while(ch>='0'&&ch<='9'){
        x=(x<<1)+(x<<3)+ch-'0';
        ch=getchar();
    }
    x*=fh;
}

char s[maxn];
int n,m,sa[maxn],x[maxn],y[maxn],ct[maxn];

int chk(int x){
    return x>0?x:x+n;
}

void SA(){
    for(register int i=1;i<=n;i++) ct[x[i]=s[i]]++;
    for(register int i=2;i<=m;i++) ct[i]+=ct[i-1];
    for(register int i=n;i>=1;i--) sa[ct[x[i]]--]=i;
    for(register int k=1;k<=n;k<<=1){
        int tot=0;
        for(register int i=n-k+1;i<=n;i++) y[++tot]=i;
        for(register int i=1;i<=n;i++) if(sa[i]>k) y[++tot]=sa[i]-k;
        for(register int i=1;i<=m;i++) ct[i]=0;
        for(register int i=1;i<=n;i++) ct[x[i]]++;
        for(register int i=1;i<=m;i++) ct[i]+=ct[i-1];
        for(register int i=n;i>=1;i--) sa[ct[x[y[i]]]--]=y[i],y[i]=0;
        swap(x,y);x[sa[1]]=tot=1;
        for(register int i=2;i<=n;i++)
            if(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) x[sa[i]]=tot;
            else x[sa[i]]=++tot;
        if(tot==n) break;
        m=tot;
    }
}

int rnk[maxn];
int let;
int main(){
    ios::sync_with_stdio(0);
    cin>>(s+1);n=strlen(s+1);let=n;
    for(register int i=n+1;i<=n*2;i++) s[i]=s[i-n];
    n*=2;
    m=122;SA();
    for(register int i=1;i<=n;i++){
        rnk[sa[i]]=i;
    }
    for(register int i=1;i<=n;i++){
        if(sa[i]<=let)
            cout<<s[sa[i]+let-1];
    }
    cout<<endl;
    return 0;
}