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

来源:互联网 发布:js如何获取input的值 编辑:程序博客网 时间:2024/05/16 02:14

题意大意:每个人写一行代码,会产生a个bug,给第总人数,需要写的代码总行数,及产生的总bug的最大值,每个人可以写多行代码(>=0),求总共有多少方案数(即安排每个人写的代码数的方案)。

连接:http://codeforces.com/contest/544/problem/C

其实是个完全背包,我就连别人的代码看了很久很久才有所觉悟,我们不难发现人数,代码数,bug数是需要我们考虑的额,首先我们只考虑人数,代码数,d【i】【j】表示前i个人写j行代码的方案数,因为一个人可以选择写多条代码,是不是就是个完全背包,d【i】【j】=d【i】【j-v[i]】+d[i-1][j];因为每个人选一次写的代码数为一,所以这里v【i】==1,然后我们考虑bug数,d【i】【j】【k】= d【i-1】【j】【k】(第i个人没写1行代码)+d【i】【j-1】【k-a[i]】(再选第i个人写,第i个人至少写了1行代码);考虑到类存限制,我们把完全背包换成滚动数组。


其实我看到三维dp就不知从何入手,但是只要抓住关键,状态转移也就出来了,此题分为两种情况,第i个人没写一行,或者再选第i个人写,第i个人至少写一行。


#include<stdio.h>#include<string.h>#include<iostream>#include<algorithm>using namespace std;long long d[510][510];int a[510];int main(){int n,m,b,mod,i,j,k;long long s=0;while(scanf("%d%d%d%d",&n,&m,&b,&mod)!=EOF){s=0;memset(d,0,sizeof(d));for(i=1;i<=n;i++)scanf("%d",&a[i]);d[0][0]=1;for(i=1;i<=n;i++){for(j=1;j<=m;j++){for(k=a[i];k<=b;k++){d[j][k]=(d[j][k]+d[j-1][k-a[i]])%mod;}}}for(k=0;k<=b;k++)s=(d[m][k]+s)%mod;printf("%I64d\n",s);}}




0 0
原创粉丝点击