题意:有一个天平,一些砝码和挂钩,要你将全部的砝码挂上使得天平平衡。
题解:因为存在负值,所以用一个一位数组可能会使某个砝码被重复使用(当然,也有人用两个一维数组来解决)。而二维数组避免了这个问题。由于砝码能够产生的最大力矩为+-7500,我们设置一个偏移量以避免数组下标为负。
#include <iostream>
using namespace std;
int c[25], g[25], dp[21][17000];
int main()
{
int C, G, v, i, j, weight = 0;
scanf("%d%d",&C,&G);
for ( i = 1; i <= C; i++ )
scanf("%d",c+i);
for ( i = 1; i <= G; i++ )
{
scanf("%d",g+i);
weight += g[i];
}
int left = c[1] * weight;
int right = c[C] * weight;
int balance = 0 - left;
memset(dp,0,sizeof(dp));
dp[0][balance] = 1;
for ( i = 1; i <= G; i++ )
{
for ( v = 0; v <= right - left; v++ )
{
if ( dp[i-1][v] != 0 )
{
for ( j = 1; j <= C; j++ )
dp[i][v+g[i]*c[j]] += dp[i-1][v];
}
}
}
printf("%d\n",dp[G][balance]);
return 0;
}