Codeforces Round #302 (Div. 2) C Writing Code(dp)

链接
n个程序员写m行代码,每个程序员写一行代码会出现ai个bug。一个优秀的plan是每个程序员写vi行代码。

使得:v1+v2+..+vn=m。并且总的bug不超过b。有多少种这样的plan。
题解:

#include<bits/stdc++.h>
using namespace  std;
#define ll long long
#define pb push_back
#define inf 2099999999
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
const int N=1e3;
ll arr[N];
ll dp[N][N];
/*
    dp[j][k]表示代码j行 出现k个bug的情况种数
    那么dp[j][k]=dp[j][k]+dp[j-1][k-arr[i]]
*/
int main()
{
    freopen("D://rush.txt", "r", stdin);
    ios::sync_with_stdio(false),cin.tie(0);
    ll n,m,b,mod,sum=0;
    cin>>n>>m>>b>>mod;
    rep(i,1,n)
    cin>>arr[i];
    dp[0][0]=1;
    /*
        这里的枚举类似于完全背包问题
        把i种物品无限放入m大小的背包中
        同时bug数不能超过b
    */
    for(int i=1;i<=n;i++)
    for(int k=arr[i];k<=b;k++)
    for(int j=1;j<=m;j++)
    {
        dp[j][k]=dp[j][k]+dp[j-1][k-arr[i]];
        dp[j][k]%=mod;
    }
    for(int k=0;k<=b;k++)
        sum+=dp[m][k],sum%=mod;
    cout<<sum<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/ffgcc/article/details/81029110
今日推荐