POJ 1037 A decorative fence(dp)

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值