POJ 1837
给出numc个位置,numg个砝码。砝码只能挂在给出的位置上且所有的砝码都必须被使用,问使天平平衡的方案有多少个。
首先要知道,天平平衡相当于:Σlength左 *weight左 + Σlength右 *weight右 = 0(题目中说了左边的位置取-)
dp数组dp[i][j]表示取到第i个筹码时,天平的平衡状态。平衡状态即 B = Σlength左 *weight左 + Σlength右 *weight右。若为0,说明天平平衡,若为负,说明向左倾斜,若为正,说明向右倾斜。
但是这里有一个问题,数组的下标不能为负。我们知道,天平平衡状态的极限就是,所有砝码都堆在一侧,并且是lengthmax的位置,也就是20 *25 * 15=7500。那么,原本j的范围应为[-7500, 7500],不妨平移7500个单位,转化为[0, 15000]。此时就可以用dp数组来表示了。平衡的方案数就是dp[numg][7500]。
这里有个问题我不太明白,我看到题解里说这是01背包问题。怎么理解?是通过g[i] *c[r]的组合,相当于有价值分别为v[++k]=g[i] *c[r]的k种物品且每件只有一件?
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int c[30],g[30],
dp[30][15010];//dp[i][j]表示到第i个砝码时的平衡状态(7500平衡)
//臂长[-15,15] 砝码范围[1,25] 个数[2,20]
//故而极限情况 15*25*20=7500
int main()
{
int numc, numg;
scanf("%d%d", &numc, &numg);
for(int i = 1;i <= numc;i++)
scanf("%d", &c[i]);//c[i] position
for(int i = 1;i <= numg;i++)
scanf("%d", &g[i]);//g[i] weight
//dp[i][j] 第i个物品时偏移量j
//memset(dp, 0, sizeof(dp));
dp[0][7500] = 1;//什么都不挂时一种方案
for(int i = 1;i <= numg;i++)
{
for(int j = 0;j <= 15000;j++)
{
if(dp[i-1][j])
{
//printf("go into loop-if\n dp[%d-1][%d]=%d\n", i, j, dp[i-1][j]);
for(int r = 1;r <= numc;r++)
{
dp[i][j+g[i]*c[r]] += dp[i-1][j];
//printf("dp[%d][%d]=%d\n",i,j+g[i]*c[r],dp[i][j+g[i]*c[r]]);
}
}
}
}
printf("%d\n",dp[numg][7500]);
return 0;
}