【BZOJ4104】解密运算 [暴力]

解密运算

Time Limit: 10 Sec  Memory Limit: 512 MB
[Submit][Status][Discuss]

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

Solution

　　显然，按照权值（即这个字符串首字母）为第一关键字，位置（即这个字符后面的串的排名）为第二关键字排序。
　　然后这时候权值就是矩阵的第一列。
　　形如：
　　　　*ABCD
　　　　-----
　　　　-----

　　这时第一行后面ABCD的排名，显然就是读入中*的位置，因为*作为ABCD*的最后一个字符。
　　比如ABCD*排名为x，那么这时候 A 显然在矩阵第一列中的第x个，输出即可。

Code

 #include<iostream>
 #include<string>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 #include<queue>
 using namespace std;
 typedef long long s64;

 const int ONE = 2e6 + ;
 const s64 INF = 1e18;

 int n, m;
 struct power
 {
         int val, pos;
 }a[ONE];

 int get()
 {
         int res=,Q=;char c;
         while( (c=getchar())< || c> )
         if(c=='-')Q=-;
         res=c-;
         while( (c=getchar())>= && c<= )
         res=res*+c-;
         return res*Q;
 }

 int cmp(const power &a, const power &b)
 {
         if(a.val != b.val) return a.val < b.val;
         return a.pos < b.pos;
 }

 int main()
 {
         n = get();    m = get();
         for(int i = ; i <= n + ; i++)
             a[i] = (power){get(), i};
         sort(a + , a + n +  + , cmp);

         int pos = ;
         for(int i = ; i <= n; i++)
         {
             pos = a[pos].pos;
             printf("%d ", a[pos].val);
         }
 }