题意:
有一个天平,天平上有若干钩子,钩子上可以挂勾码。现天平上有C个钩子,G个勾码,每个勾码的重量已知。求将勾码全部挂到钩子上且使天平平衡的方法种数。
思路:
利用动态规划求解。
假设表征天平倾斜度的量=重量*距天平中心点的距离,中心点左边距离为负,中心点右边距离为正。
定义dp[i][j]表示在挂了前i个重物,天平倾斜度为j的放置方法总数。
c[i]记录钩子距天平中心点的距离。
w[i]记录第i个钩子的重量(从0开始)。
若挂了前i - 1个重物,平衡度为j的方法数已知,即dp[i - 1][j]已知
那么,挂了前i个重物,平衡度为 j + c[k]*w[i - 1]的方法数即为
dp[i][j + c[k]*w[i - 1]] + dp[i - 1][j];
则有:
dp[i][j + c[k]*w[i - 1]] += dp[i - 1][j];
在dp[i][j]中,j的范围为-7500-7500。为了不让数组下标出现负数,以天平最左端的点为原点,将j的范围平移到0-15000。
初始化时,dp[0][7500] = 1,即什么都不挂。
代码如下:
#include <iostream>
using namespace std;
int dp[21][15001] = {0}; //挂满前i个重物,平衡度为j时的解决方案个数
int main()
{
int n = 0; //钩子数量
int g = 0; //重物数量
int c[21] = {0}; //钩子位置
int w[21] = {0}; //重物重量
cin>>n>>g;
for(int i = 0; i < n; i++)
{
cin>>c[i];
}
for(int i = 0; i < g; i++)
{
cin>>w[i];
}
dp[0][7500] = 1; //挂0个重物,平衡度为7500的解决方案个数为1
for(int i = 1; i <= g; i++)
{
for(int j = 0; j <= 15000; j++)
{
for(int k = 0; k < n; k++) //遍历k个位置
{
dp[i][j + c[k]*w[i - 1]] += dp[i - 1][j];
}
}
}
cout << dp[g][7500] <<endl;
return 0;
}