GYM 101086 D.Secure but True（水~）

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;
}