The Preliminary Contest for ICPC Asia Shanghai 2019 -F. Rhyme scheme(DP+记录路径)

题目链接：https://nanti.jisuanke.com/t/41414

题意：

输出长度为n的第k小的Rhyme scheme字符串.
Rhyme scheme: 长度为n的个数是bell number的例子
Rhyme scheme字符串：目前的字母最大是前面出现过的字母+1，即AABC是符合的，AACB是不符合的，因为C前面没有出现B。
题目要求输出长度为n的字典序第k小的 Rhyme scheme字符串。
有一万个样例，每个样例输入一个n和k，表示长度为n的第k小的字符串。

思路：

可以发现这种字符串构成了一颗字符数，即根节点是A，第二层是AB，第三层是AB，ABC……由于数据比较大（long long）都存不下，所以不可能直接在树上搜索。用dp[i][j]表示目前字母为i，后面还有j位的时候可以构成的字符串的数量。那么就有递推公式：
dp[i][j]=i* dp[i][j-1]+dp[i+1][j-1];
意思就是下一位可以取A~A+i+1中的字母，其中A ~A+i中每一个字母的方案数都是dp[i][j-1]，所以是i*dp[i][j-1]，而A+i+1这个字符的方案数是dp[i+1][j-1]。
查询时只需要按照从小到大的顺序比较k与dp[][]D的大小，然后用ans数组记录第i为选取的字母，那么就可以得出正确答案。由于T很大，所以需要先预处理出来的dp数组。
注意需要开__int128，因为dp[1][26]的大小已经超过了 ull 。

AC代码：

#include<bits/stdc++.h>
#include <stdlib.h>
#include<algorithm>
#include<complex>
#include<iostream>
#include<iomanip>
#include<ostream>
#include<cstring>
#include<string.h>
#include<string>
#include<cstdio>
#include<cctype>
#include<vector>
#include<queue>
#include<set>
#include<stack>
#include<map>
#include<cstdlib>
#include<time.h>
#include<ctime>
#include<bitset>
// #include<ext/pb_ds/assoc_container.hpp>
// #include<ext/pb_ds/hash_policy.hpp>
using namespace std;
// using namespace __gnu_pbds;
#define pb push_back
#define _filein freopen("C:\\Users\\zsk18\\Desktop\\in.txt","r",stdin)
#define _fileout freopen("C:\\Users\\zsk18\\Desktop\\out.txt","w",stdout)
#define ok(i) printf("ok%d\n",i)
#define gcd(a,b) __gcd(a,b) ;
typedef double db;
//typedef long long ll;
typedef __int128 ll;
typedef unsigned long long ull;
typedef pair<int,int>PII;
const double PI = acos(-1.0);
const ll MOD=1e9+7;
const ll NEG=1e9+6;
const int MAXN=50;
const int INF=0x3f3f3f3f;
const ll ll_INF=1e15;
const double eps=1e-9;
ll qm(ll a,ll b){ll ret=1;while(b){if(b&1)ret=ret*a%MOD;a=a*a%MOD;b>>=1;}return ret;}
ll inv(ll x){return qm(x,MOD-2);} 
ll dp[MAXN][MAXN];
ll ans[MAXN];
int n;
char s[MAXN];
int main(void)
{
      dp[26][1]=26,dp[26][0]=1;
      for(int i=25;i;i--)dp[i][1]=i+1,dp[i][0]=1;
      for(int j=2;j<=26;j++)
      {
            for(int i=25;i;i--)
            {
                  dp[i][j]=(ll)i*dp[i][j-1]+dp[i+1][j-1];
                 if(dp[i][j]>5e19)
                        dp[i][j]=(ll)5e19;
            }
      }
      // for(int i=1;i<=26;i++)
      //       for(int j=0;j<=10;j++)
      //             printf("dp[%d][%d]=%I64d\n",i,j,dp[i][j]);
      int t;
      scanf("%d",&t);
      int number=0;
      while(t--)
      {
            scanf("%d%s",&n,s);
            ll k=0;
            int len=strlen(s);
            for(int i=0;i<len;i++)
                  k=k*10+(ll)(s[i]-'0');
            int ma=1;
            for(int j=1;j<=n;j++)
            {
                  for(int i=1;i<=26;i++)
                  {
                        ma=max(ma,i);
                        if(k<=dp[ma][n-j])
                        {
                              ans[j]=i;
                              break;
                        }
                        k-=dp[ma][n-j];
                  }
            }
            printf("Case #%d: ",++number);
            for(int i=1;i<=n;i++)
                  printf("%c",'A'+(int)ans[i]-1);
            printf("\n");

      }

      return 0;
}