背包DP || Codeforces 544C Writing Code

Posted 2021-01-02 pinkglightning

程序员写bug的故事23333

题意：n个程序员，一共写m行程序，最多产生b个bug，问方案数

思路：f[i][j]表示写了i行，产生了j个bug的方案数，因为每个人都是可以独立的，所以i循环到n都做一遍 f[i][j] += f[i-1][j-a[i]] 在前一行  i 的 a[i] 个bug还没有写上去的情况数

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
int f[550][550], a[550];

int main()
{
    int n, m, b, mod;
    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[j][k] = (f[j][k] + f[j-1][k-a[i]]) % mod;
    LL ans = 0;
    for(int i = 0; i <= b; i++) ans = (ans + f[m][i]) % mod;
    printf("%lld
", ans);
    return 0;
}