DP起手练习3

题目描述：

CF543A码代码

有n个程序员，每个程序员都可以写任意行代码，总共要编写m行代码，这m行代码可以由多个程序员来编写。但是第i个程序员在一行代码中会出现$a_i$个 bug。现在希望知道有多少种方案能使得这m行代码中的 bug 的数量不超过b个。
两个方案不同当且仅当某个程序员编写的代码量（行数）不同。

输入格式：

输入第一行包含三个整数 n,m,b,mod。
接下来一行n个整数$a_i$。

输出格式：

输出一行一个整数，表示m行代码 bug 数量小于 b的方案数对 mod取模后的答案.$1\le n,m\le 500,0\le b\le 500;1\le mod\le 10^9+7;0\le a_i\le 500$.

输入输出样例

输入 #1
3 3 3 100
1 1 1
输出 #1
10
输入 #2
3 6 5 1000000007
1 2 3
输出 #2
0
输入 #3
3 5 6 11
1 2 1
输出 #3
0

思路

初看觉得对DP新手很不有好,但经过仔细分析后发现：这TM 不就是背包吗？然后就很容易得到DP转移方程:
($f[k][j]表示写j行代码出现k个bug的方案总数$)
$f[k][j]=f[k][j]+f[k-a[i]][j-1](a[i]为第i位程序员写1行代码出现的bug个数)$
最后答案为$\sum f[i][m](0<=i<=b,i\in z)$.(注意取模!)

代码

#include<bits/stdc++.h>
#define N 505
using namespace std;

int n,m,b,mod,ans=0;
int a[N],f[N][N];

int main()
{
	scanf("%d%d%d%d",&n,&m,&b,&mod);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	f[0][0]=1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			for(int k=a[i];k<=b;k++)
			f[k][j]=(f[k-a[i]][j-1]+f[k][j])%mod;//是背包吧？
	for(int i=0;i<=b;i++)
	ans=(ans+f[i][m])%mod;
	printf("%d\n",ans%mod);
	return 0;
}

​