BZOJ 2287: 【POJ Challenge】消失之物

Description

ftiasch 有 N 个物品, 体积分别是 W₁, W₂, ..., W_N。 由于她的疏忽, 第 i 个物品丢失了。 “要使用剩下的 N - 1 物品装满容积为 x 的背包，有几种方法呢？” -- 这是经典的问题了。她把答案记为 Count(i, x) ，想要得到所有1 <= i <= N, 1 <= x <= M的 Count(i, x) 表格。

Input

 

第1行：两个整数 N (1 ≤ N ≤ 2 × 10³) 和 M (1 ≤ M ≤ 2 × 10³)，物品的数量和最大的容积。

第2行： N 个整数 W₁, W₂, ..., W_N, 物品的体积。

Output

 

一个 N × M 的矩阵， Count(i, x)的末位数字。

Sample Input

3 2
1 1 2

Sample Output

11
11
21

HINT

如果物品3丢失的话，只有一种方法装满容量是2的背包，即选择物品1和物品2。

题解

背包dp，f[i][j]表示前i个物品放满j容量的包的方案数，c[i][j]表示不用i放满j的方案。1、当j<a[i]时，c[i][j]=f[n][j]。2、当j>=a[i]时，c[i][j]=f[n][j]-c[i][j-a[i]]。3、由2可知c[i][0]=1.

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
int n,m,a[2002];
ll f[2002][2002],c[2002][2002];
void init()
{
	scanf("%d%d",&n,&m);
	int i;
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
}
void dp()
{
	int i,j,k;
	f[0][0]=1;
	for(i=1;i<=n;i++)
	for(j=0;j<=m;j++)
	   {f[i][j]=(f[i][j]+f[i-1][j])%10;
	    if(j>=a[i]) f[i][j]=(f[i][j]+f[i-1][j-a[i]])%10;
	   }
	
	for(i=1;i<=n;i++)
	   {for(j=0;j<=m;j++)
	       {if(j<a[i]) c[i][j]=f[n][j];
	        else c[i][j]=(f[n][j]-c[i][j-a[i]]+10)%10;
	        if(j>0) printf("%lld",c[i][j]);
		   }
		printf("\n");
	   }
}
int main()
{
	init(); dp();
	return 0;
}