題目大意:有一個天平, 平衡臂上有C個鉤子, (ci < 0 表示在左邊, ci > 0 表示鉤子在右邊, 值表示距離終點的長度)。 然後是G個砝碼的重量, 要求所有的砝碼都用上,求使天平平衡的方案有多少種。
解題思路:W[i] 表示第i個砝碼的重量, L[i]保存每個鉤子的位置。
要求掛i個砝碼平衡時有多少中, 那就是加上這個砝碼之前這個臂上少多少力。 dp[i][j] 表示掛上i個砝碼的時候的狀態, j<0 表示向右偏, 即左邊缺少多少牛力。 j>0即表示右邊缺少多少牛的力, 由於數組下標不能是負的, 我們可以讓最小的情況爲0, 則中點即 最大的可能 mid = 20 * 20 * 15; 右邊的都大於mid, 左邊的都小於 mid。
因此狀態轉移方程:dp[i][j + w[i] * L[k]] = dp[i-1][j]
代碼:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int mid = 20 * 20 * 15;
const int rig = mid*2;
int dp[25][rig + 5];
int L[25], W[25];
int main()
{
int n, m;
while(~scanf("%d%d", &n, &m))
{
for(int i=0; i<n; ++i) scanf("%d", &L[i]);
for(int i=0; i<m; ++i) scanf("%d", &W[i]);
memset(dp, 0, sizeof dp);
dp[0][mid] = 1;
for(int i=1; i<=m; ++i)
for(int k = 0; k<n; ++k)
for(int j = 0; j <= rig; ++j)
//if(j + W[i]*L[k] >=0 && j + W[i]*L[k] <= rig)
if(dp[i-1][j])
dp[i][j+W[i-1]*L[k]] += dp[i-1][j];
cout << dp[m][mid] <<endl;
}
return 0;
}