题意:输出长度为 n 的第 k 个 Rhyme scheme。Rhyme scheme 就是第 i 个字母最多是前面出现过的字母 +1
分析:
这里借用官方题解的一张图
所有的 Rhyme scheme 构成了一棵字典树,设 dp[ n ][ i ][ j ] 为长度为 n 的 Rhyme scheme 到第 i 位且出现过的最大字母为 j 的序列数。则求解第 k 个就可以用比较法填字母(可以先略过这句话,具体看代码)
从下往上推,有
不难理解,若当前第 i 层最大字母为 j ,第 i+1 层最大字母也为 j ,则第 i+1 层只能填 1~j ,一共 j 种选择,但若是第 i+1 层最大字母是 j+1,则第 i+1 层只能填 j+1。
B(26)会爆long long,所以得用 __int128 或者 合两个long long (肯定是用 __int128 啦,方便又省事 []~( ̄▽ ̄)~* )
代码:
#include<cstdio>
#include<iostream>
using namespace std;
__int128 dp[30][30][30];
void read(__int128 &x)
{
x=0;
int f = 1;
char ch=getchar();
while(!(ch>='0'&&ch<='9')) ch=getchar();
x = x*10 + ch-'0';
while((ch = getchar()) >= '0' && ch <= '9')
x = x*10 + ch-'0';
x *= f;
}
int main()
{
for(int n=1;n<=26;n++)
for(int i=n;i>=1;i--)
{
if(i==n)
{
for(int j=1;j<=i;j++) dp[n][i][j]=1ll;
}
else
{
for(int j=1;j<=i;j++)
dp[n][i][j]=dp[n][i+1][j]*j+dp[n][i+1][j+1];
}
}
int T,n;
__int128 k;
scanf("%d",&T);
for(int cas=1;cas<=T;cas++)
{
scanf("%d",&n);
read(k);
printf("Case #%d: ",cas);
int t=0,c;
for(int i=1;i<=n;i++)
{
/*
t 为这一层之前出现过的最大字母,则这一层的字母上限是 t+1
如果k > dp[n][i][t],则表明若以当前c作为第i位的字母,接下
来的所有满足条件的Rhyme scheme的数量和小于k,则证明第 i
位填的字母肯定比c大
*/
for(c=1;c<t+1&&k>dp[n][i][t];c++)
{
k-=dp[n][i][t];
}
putchar('A'+c-1);
t=max(t,c);
}
puts("");
}
return 0;
}