Description
合法串为只由镜面对称的字符{‘A’,’H’,’I’,’M’,’O’,’T’,’U’,’V’,’W’,’X’,’Y’}组成,把所有合法串首先按长度升序排,相同长度按字典序升序排,给出一个合法串,求按顺序在这个串后面第k个合法串
Input
第一行一整数T表示用例组数,每组用例输入一整数k和一字符串s(1<=T<=256,1<=k<=1e9,1<=|s|><=1e3)
Output
对于每组用例,输出s后第k个字符串
Sample Input
4
2 M
4577 ATM
491 AI
132 A
Sample Output
T
ITMO
MAX
AAA
Solution
将串反转后可以看做一个11进制的数字,首先判s距长度为|s|+1的第一个串还差多少,如果k比这个差值大就先补成|s|+1个A,然后进行同样的步骤,如果不够那就从最低位开始进位即可
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 1111
char c[]={'A','H','I','M','O','T','U','V','W','X','Y'},s[maxn];
int T,k,n,a[111];
void solve()
{
ll temp=0;
int flag=0;
for(int i=n-1;i>=0;i--)
{
temp=11ll*temp+10-a[s[i]];
if(temp>=k)
{
flag=1;
break;
}
}
if(!flag)
{
k-=temp+1;
n++;
for(int i=0;i<n;i++)s[i]='A';
solve();
}
else
{
int up=0;
for(int i=0;i<n;i++)
{
up+=a[s[i]]+k%11;
s[i]=c[up%11];
up/=11,k/=11;
}
for(int i=n-1;i>=0;i--)printf("%c",s[i]);
printf("\n");
}
}
int main()
{
for(int i=0;i<11;i++)a[c[i]]=i;
scanf("%d",&T);
while(T--)
{
scanf("%d %s",&k,s);
n=strlen(s);
for(int i=0,j=n-1;i<j;i++,j--)swap(s[i],s[j]);
solve();
}
return 0;
}