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=;char ch=;int fh;
     while(ch!='-'&&(ch<''||ch>'')) ch=getchar();
     if(ch=='-') fh=-,ch=getchar();
     else fh=;
     while(ch>=''&&ch<=''){
         x=(x<<)+(x<<)+ch-'';
         ch=getchar();
     }
     x*=fh;
 }

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

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

 void SA(){
     for(register int i=;i<=n;i++) ct[x[i]=s[i]]++;
     for(register int i=;i<=m;i++) ct[i]+=ct[i-];
     for(register int i=n;i>=;i--) sa[ct[x[i]]--]=i;
     for(register int k=;k<=n;k<<=){
         int tot=;
         for(register int i=n-k+;i<=n;i++) y[++tot]=i;
         for(register int i=;i<=n;i++) if(sa[i]>k) y[++tot]=sa[i]-k;
         for(register int i=;i<=m;i++) ct[i]=;
         for(register int i=;i<=n;i++) ct[x[i]]++;
         for(register int i=;i<=m;i++) ct[i]+=ct[i-];
         for(register int i=n;i>=;i--) sa[ct[x[y[i]]]--]=y[i],y[i]=;
         swap(x,y);x[sa[]]=tot=;
         for(register int i=;i<=n;i++)
             if(y[sa[i]]==y[sa[i-]]&&y[sa[i]+k]==y[sa[i-]+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();
     cin>>(s+);n=strlen(s+);let=n;
     for(register int i=n+;i<=n*;i++) s[i]=s[i-n];
     n*=;
     m=;SA();
     for(register int i=;i<=n;i++){
         rnk[sa[i]]=i;
     }
     for(register int i=;i<=n;i++){
         if(sa[i]<=let)
             cout<<s[sa[i]+let-];
     }
     cout<<endl;
     return ;
 }