Description
给定长度依次为1到n的木棒n个, 摆放规则为除了两边的木棒,剩下的木棒必须要比相邻的两个都高或者都低,求字典序第c小的序列
Input
第一行一整数T表示用例组数,每组用例两个整数n,c(T<=100,N<=20)
Output
对于每组用例,输出所有合法序列中字典序第c小的序列
Sample Input
2
2 1
3 3
Sample Output
1 2
2 3 1
Solution
设dp[i][j][0]为长度为j的序列,j个数中第i小的数在第一位(就是从小到大排序后第i个位置的数),且先升再降的方法数,dp[i][j][1]是先降再升的方法数,那么轻松得到递推方程:
dp[i][j][0]=sum{dp[k][j-1][1],k=i,…,j},i=1,2,…,j
dp[i][j][1]=sum{dp[k][j-1][0],k=1,…,i-1},i=1,2,…,j
这里解释一下式子中k的范围,第一个式子中dp[i][j][0]表示从这j个数中已经取出了第i小的,那么在剩下的j-1个数中,从第i小的到第j-1小的数全部都大于i,所以k的取值是i~j,第二个式子同理
至于如何输出字典序第c小的序列,就是枚举n位的每一位即可,但是由于dp数组第一位是第i小的而不是i,所以输出的时候也要稍微处理一下,就是从还没用的数中选出第ans[i]小的
Code
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
#define maxn 22
int T,n,ans[maxn],vis[maxn];
ll c,dp[maxn][maxn][2];
void init()
{
memset(dp,0,sizeof(dp));
dp[1][1][0]=dp[1][1][1]=1;
for(int j=2;j<=20;j++)
for(int i=1;i<=j;i++)
{
for(int k=i;k<j;k++)
dp[i][j][0]+=dp[k][j-1][1];
for(int k=1;k<i;k++)
dp[i][j][1]+=dp[k][j-1][0];
}
}
int main()
{
init();
scanf("%d",&T);
while(T--)
{
scanf("%d%lld",&n,&c);
int flag,cur,res=1;
ll sum=0;
for(int i=1;i<=n;i++)
{
if(sum+dp[i][n][0]+dp[i][n][1]>=c)
{
c-=sum;
if(dp[i][n][1]>=c)flag=1;
else c-=dp[i][n][1],flag=0;
ans[1]=i;
cur=i;
break;
}
sum=sum+dp[i][n][0]+dp[i][n][1];
}
n--;
while(n)
{
if(flag)
{
for(int i=1;i<cur;i++)
{
if(dp[i][n][0]>=c)
{
ans[++res]=i;
cur=i;
break;
}
c-=dp[i][n][0];
}
}
else
{
for(int i=cur;i<=n;i++)
{
if(dp[i][n][1]>=c)
{
ans[++res]=i;
cur=i;
break;
}
c-=dp[i][n][1];
}
}
flag^=1,n--;
}
n=res;
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(!vis[j])
{
ans[i]--;
if(ans[i]==0)
{
printf("%d%c",j,i==n?'\n':' ');
vis[j]=1;
break;
}
}
}
return 0;
}