Codeforces 543 A Writing Code 计数DP

来源：互联网发布：与sqlserver建立连接编辑：程序博客网时间：2024/04/30 04:22

http://codeforces.com/problemset/problem/543/A

题目大意：n个人，一共要写m行程序，每个程序员每行出现的bug数为ai，要求整个程序出现的bug数不超过b的方案数.

题目分析：简单的计数DP，完全背包。需要记录的是写到第i行时错误的bug数为j的方案。所以定义二维数组dp[i][j]。具体见代码；初始化dp[0][i]=1;(用于一开始的转移）

#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<algorithm>#include<functional>#include<cmath>#include<cctype>#include<cassert>#include<climits>using namespace std;#define For(i,n) for(int i=1;i<=n;i++)#define Rep(i,n) for(int i=0;i<n;i++)#define Fork(i,k,n) for(int i=k;i<=n;i++)#define ForD(i,n) for(int i=n;i;i--)#define Forp(x) for(int p=pre[x];p;p=next[p])#define RepD(i,n) for(int i=n;i>=0;i--)#define MEM(a) memset(a,0,sizeof(a))#define MEMI(a) memset(a,127,sizeof(a))#define MEMi(a) memset(a,128,sizeof(a))#define INF (2139062143)#define phiF (1000000006)#define MAXN (1000000+10)typedef long long ll;int n,m,b,mo,ai[1000],dp[505][505];int main(){scanf("%d%d%d%d",&n,&m,&b,&mo);For (i,n) scanf("%d",&ai[i]);   MEM(dp);   Rep(i,b+1) dp[0][i] = 1;   For(k,n) {    For (i,m)        Fork (j,ai[k],b) {                   dp[i][j] = (dp[i-1][j-ai[k]] + dp[i][j]) % mo;        }   }  printf("%d",dp[m][b]);}

1 0