Balance POJ - 1837 計數dp

傳送門


題目大意:有一個天平, 平衡臂上有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;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章