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

题意:

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

题解:

1.将原串扩展成2倍(避免有些没有遍历到)。
2.建立后缀数组,对于 s a i sa_{i} sai小于 n n n的输出第 s a i + n − 1 sa_i+n-1 sai+n1位即可。

代码:

// luogu-judger-enable-o2
#include
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;
}

你可能感兴趣的:(后缀数组)