JSOI2007:字符加密(后缀数组)

Posted on | Post modified:

JSOI2007:字符加密(后缀数组)

题意：

将长度为$n$的字符串排成一圈，对于$n$种排列方式按字典序从小到大输出最后一位。

题解：

1.将原串扩展成2倍（避免有些没有遍历到）。
2.建立后缀数组，对于$sa_{i}$小于$n$的输出第$sa_i+n-1$位即可。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// luogu-judger-enable-o2
#include<bits/stdc++.h>
using namespace std;
const int N = 4e5+100;
const int mod = 1e9+7;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const ll llINF = 0x3f3f3f3f3f3f3f3f;
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define fep(i,a,b) for(int i=(a);i>=(b);i--)
inline bool read(ll &num) {
    char in;bool IsN=false;
    in=getchar();
    if(in==EOF) return false;
    while(in!='-'&&(in<'0'||in>'9')) in=getchar();
    if(in=='-'){ IsN=true;num=0;}
    else num=in-'0';
    while(in=getchar(),in>='0'&&in<='9'){
            num*=10,num+=in-'0';
    } 
    if(IsN) num=-num;
    return true;
}
string s;
int ran[N],tmp[N],sa[N];
int n,k;
bool cmp(int i,int j){
    if(ran[i]!=ran[j]) return ran[i]<ran[j];
    int ri=i+k<=n?ran[i+k]:-1;
    int rj=j+k<=n?ran[j+k]:-1;
    return ri<rj;
}
void construct_sa(string s,int *sa){
    n=(int)s.length();
    for(int i=0;i<=n;i++){
        sa[i]=i;
        ran[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k*=2){
        sort(sa,sa+n+1,cmp);
        tmp[sa[0]]=0;
        for(int i=1;i<=n;i++){
            tmp[sa[i]]=tmp[sa[i-1]]+(cmp(sa[i-1],sa[i])?1:0);
        }
        for(int i=0;i<=n;i++) ran[i]=tmp[i];
    }
}
int main(){
   // freopen("1.in", "r", stdin);
    while(getline(cin,s)){
        int l=(int)s.length();
        s=s+s;
        construct_sa(s,sa);
        rep(i,1,n){
            if(sa[i]<=l-1) cout<<s[sa[i]+l-1];
        }
        cout<<endl;
    }
    return 0;
}