洛谷 4051 [JSOI2007]字符加密（后缀数组）

题目描述

喜欢钻研问题的JS 同学，最近又迷上了对加密方法的思考。一天，他突然想出了一种他认为是终极的加密办法：把需要加密的信息排成一圈，显然，它们有很多种不同的读法。

例如‘JSOI07’，可以读作： JSOI07 SOI07J OI07JS I07JSO 07JSOI 7JSOI0 把它们按照字符串的大小排序： 07JSOI 7JSOI0 I07JSO JSOI07 OI07JS SOI07J 读出最后一列字符：I0O7SJ，就是加密后的字符串（其实这个加密手段实在很容易破解，鉴于这是突然想出来的，那就^^）。但是，如果想加密的字符串实在太长，你能写一个程序完成这个任务吗？

输入输出格式

输入格式：

输入文件包含一行，欲加密的字符串。注意字符串的内容不一定是字母、数字，也可以是符号等。

输出格式：

输出一行，为加密后的字符串。

输入输出样例

输入样例#1：复制

JSOI07

输出样例#1：复制

I0O7SJ

说明

对于40%的数据字符串的长度不超过10000。

对于100%的数据字符串的长度不超过100000。

题解：思考一下好像环不是很好搞，所以索性拆开来，变成原来的两倍长度，显然一个后缀的前n个字母就是会读作的字符串，而且对两倍长度的字符串搞后缀排序显然是可以的

所以就拆开后后缀排序一下，只统计sa[i]<=n的情况就可以了

代码如下：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

int n,k,sa[300010],rk[300010],tmp[300010];
char s[300010];

int cmp(int i,int j)
{
    if(rk[i]!=rk[j]) return rk[i]<rk[j];
    int ri=i+k<=n?rk[i+k]:-1;
    int rj=j+k<=n?rk[j+k]:-1;
    return ri<rj;
}

int main()
{
    scanf("%s",s);
    n=strlen(s);
    for(int i=n;i<=n*2-1;i++)
    {
        s[i]=s[i-n];
    }
    n=2*n;
    for(int i=0;i<=n;i++)
    {
        sa[i]=i;
        rk[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k<<=1)
    {
        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]);
        }
        for(int i=0;i<=n;i++)
        {
            rk[i]=tmp[i];
        }
    }
    sort(sa,sa+n+1,cmp);
    string ans="";
    for(int i=1;i<=n;i++)
    {
        if(sa[i]+n/2<n) ans+=s[sa[i]+n/2-1];
    }
    cout<<ans<<endl;
}