题意:给 出 n,k<=20 , 0<=L<=10^9. 要求构造一个序列a_1,a_2,...,a_n,满足 0<=a_i<=l,并且其中的某些数的sum=k,求可能的序列的总数。
dp[p] 表示当前构造的序列可以加和得到(1,2,3...k的各种组合)的方案总数。
状态转移不难想,一个状态p ,在当前序列的末位加上数字j,新的状态是 newp=(p|((p<<j)&((1<<k)-1))|(1<<(j-1)));
可见 新的状态 newp>=p ,可以减一维空间。
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
const int mod =1000000007;
int n,k,l;
int x;
long long dp[1<<22];
int t;
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d",&t);
while (t--){
scanf("%d%d%d",&n,&k,&l);
if (l>k) x=l-k;
else x=0;
l=min(l,k);
memset(dp,0,sizeof(dp));
dp[0]=1;
for (int i=0;i<n;i++){
for (int p=(1<<k);p>=0;p--)
if (dp[p]>0) {
int yy=dp[p];
dp[p]=(dp[p]*x)%mod;
for (int j=0;j<=k;j++){
int newp;
if (j==0) newp=p;
else newp=(p|((p<<j)&((1<<k)-1))|(1<<(j-1)));
dp[newp]+=yy;
if (dp[newp]>=mod) dp[newp]-=mod;
}
}
}
int ans=0;
for (int p=(1<<(k-1));p<(1<<k);p++)
{
ans+=dp[p];
if (ans>=mod) ans-=mod;
}
cout<<ans<<endl;
}
return 0;
}