题面:
N - 这是一道比CCCC简单题更有想象力的中档题
Time Limit: 9000/3000MS (Java/Others) Memory Limit: 65535/65535KB (Java/Others)
在2550年,CCCC已经成为了一场和ICPC一样正式的大型比赛
UESTC在那年派出了一只n个人组成的参赛队参加比赛
2050年的CCCC赛制有了改变,这n个人可以坐在一起讨论问题,一起写代码!
CCCC嘛,前面都是些不着边际的奇怪题目,一直到了最后的终极大BOSS题目
最后大boss的AC代码,将由很多队员一起写成!
已知n个参赛队员,对于第i个队员,每写一行代码,就会留下 ai ai 个bug
最后一题需要写m行代码,请安排各个队员写的代码行数(显然要非负),使得整个代码的bug数不超过b个
然后,在ACM玄学之神的保佑下,这份不超过b个bug的代码就能AC了!
问你有多少种不同的安排方案可以写出一份AC代码,要求方案数对mod取模
Input
第一行,n,m,b,mod,1<=n,m<=500,0<=b<=500,1<=mod<=1e9+7
接下来一行n个数,分别是 a1 a1到 an, an, 0<= ai ai <=500
Output
对mod取模之后的方案数
Sample input and output
Sample Input | Sample Output |
---|---|
3 3 3 100 1 1 1 | 10 |
一个很正常的想法:
二维dp定义,即dp[i][j]表示当前分配了I行代码出去了,有j个bug
那么只需要枚举下分配的代码行数
然后直接按照一维完全背包的样子转移就好
AC代码(500^3)
#include<bits/stdc++.h>
using namespace std;
const int N=600;
int dp[N][N],bug[N],n,m,MAX,mod;
int main()
{
//freopen("in.in","r",stdin);
scanf("%d%d%d%d",&n,&m,&MAX,&mod);
for(int i=1;i<=n;i++)scanf("%d",&bug[i]);
memset(dp,0,sizeof(dp));
dp[0][0]=1%mod;
for(int i=1;i<=n;++i)
{
for(int j=0;j<m;++j)
for(int k=0;k+bug[i]<=MAX;++k)
{
if(!dp[j][k])continue;
dp[j+1][k+bug[i]]+=dp[j][k],dp[j+1][k+bug[i]]%=mod;
}
}
int ans=0;
for(int i=0;i<=MAX;i++)
ans+=dp[m][i],ans%=mod;
cout<<(ans%mod+mod)%mod;
}