BZOJ 4104: [Thu Summer Camp 2015]解密运算

Description

对于一个长度为N的字符串，我们在字符串的末尾添加一个特殊的字符”.”。之后将字符串视为一个环，从位置1,2,3,…,N+1为起点读出N+1个字符，就能得到N+1个字符串。

比如对于字符串“ABCAAA”，我们可以得到这N+1个串：
ABCAAA.
BCAAA.A
CAAA.AB
AAA.ABC
AA.ABCA
A.ABCAA
.ABCAAA
接着我们对得到的这N+1个串按字典序从小到大进行排序（注意特殊字符“.”的字典序小于任何其他的字符）结果如下：
.ABCAAA
A.ABCAA
AA.ABCA
AAA.ABC
ABCAAA.
BCAAA.A
CAAA.AB
最后，将排序好的N+1个串的最后一个字符取出，按照顺序排成一个新的字符串，也就是上面这个表的最后一列，就是加密后的密文“AAAC.AB”。
请通过加密后的密文求出加密前的字符串。

Input

第一行有两个整数N,M，分别表示加密前的字符串长度和字符集大小，其中字符用整数1,2,3,…,M编号，添加的特殊字符“.”用0编号。
第二行为N+1个整数，表示加密后的字符串。

Output

输出仅一行，包含N个整数，用空格隔开，依次表示加密前字符串中每个字符的编号。

Sample Input

6 3

1 1 1 3 0 1 2

Sample Output

1 2 3 1 1 1

HINT

N,M<=200000

分析

显然比较两个字符串的方法为：若第一个字符不相同，则可以直接确定，否则就看剩下两个字符串的大小。
那么如果我们以原序列第i个位置的数为开头，那么第一个字符显然就是原序列第i个字符，而剩下部分的字符串的排名显然就是i。那么我们可以以字符为第一关键字，编号为第二关键字，排序后便可以得出以每个位置为开头时的字符串排名。
那么我们可以从排名为1的位置开始，显然可以知道它后面一个数为开头的字符串的排名，从而可以知道后面一个数是多少。如此类推，便可以得到所有数。

代码

#include <bits/stdc++.h>

#define N 200005

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

std::pair <int,int> num[N];

int main()
{
    int n = read(), m = read();
    for (int i = 0; i <= n; i++)
    {
        num[i].first = read();
        num[i].second = i;
    }
    std::sort(num, num + n + 1);
    int tmp = num[0].second;
    for (int i = 1; i <= n; i++)
    {
        printf("%d ",num[tmp].first);
        tmp = num[tmp].second;
    }
    printf("\n");
}