hdu 4906 Our happy ending(状压dp)

最新推荐文章于 2017-07-26 18:34:02 发布

qian99

最新推荐文章于 2017-07-26 18:34:02 发布

阅读量731

点赞数

分类专栏： dp 文章标签： dp

本文链接：https://blog.csdn.net/qian99/article/details/38337981

版权

dp 专栏收录该内容

96 篇文章 0 订阅

订阅专栏

题意：要选择n个数组成一个集合，每个数的范围是0~L，要求从集合中选择若干个数求和可以得到k，求这样的集合的方案数。

思路：可以发现对于求和等于k有影响的只有1~k，其他的数即使选了，也不会用到，因此用dp[i]表示状态为i的情况下可以得到的方案数。i表示当前1~k这k个数是否可以得到，dp进行n次即可得到最后结果。dp时，枚举选择的数(1~min(k,L))，然后将可以得到的新的数和原来的状态或起来，最后将所有可以得到k的状态的值加起来就好了~

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int mod=1000000000+7;
int dp[1<<20];
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,k,L;
        scanf("%d%d%d",&n,&k,&L);
        ll ex=1;
        if(L>k) ex+=L-k;
        L=min(L,k);
        memset(dp,0,sizeof(dp));
        dp[0]=1;
        int msk=(1<<k)-1;
        for(int i=0;i<n;++i)
        {
            for(int j=(1<<k)-1;j>=0;--j)
            {
                int val=dp[j],state;
                if(dp[j]==0) continue;
                dp[j]=0;
                for(int x=1;x<=L;++x)
                {
                    state=j|((j<<x)&msk)|(1<<(x-1));
                    dp[state]+=val;
                    if(dp[state]>=mod) dp[state]-=mod;
                }
                dp[j]+=ex*val%mod;
                if(dp[j]>=mod) dp[j]-=mod;
            }
        }
        int ans=0;
        for(int i=0;i<(1<<k);++i)
            if(i&(1<<(k-1)))
            {
                ans+=dp[i];
                if(ans>=mod) ans-=mod;
            }
        printf("%d\n",ans);
    }
    return 0;
}