POJ 1837 Balance(二维dp)

题意:

一个天平,给n个不同长度(g[i])方向的钩子,m个不同重量(w[i])砝码,问有多少钟方法在使用了所有砝码后使得天平平衡.

思路:

一开始想的 以为钩子也要全部用上,这样就比较简单了,不需要考虑每个钩子情况,统一求正负钩子比例就好,但是交上去错的,去网上看了下才知道其实不用使用全部的钩子.

用二维数组来记录状态,dp[i][j]i表示当前以挂上第i个砝码,j表示当前的平衡状态.

平衡状态:

由于要使天平平衡,必须让两边的力矩相等,力矩=力臂*力.即相当于钩子长度*砝码重量.

由于题目用正负状态表示左右,不能记录负的状态,所以将记录状态从-7500到7500提高到0到15000,以7500作为平衡点.

可能这里就有疑问了,怎么以7500作为平衡点,接下来会讲到.

dp[i-1][j]表示使用i-1个物品时,达到j状态的方案数量,

则dp[i][j+w[i]*g[k]]+=dp[i-1][j]的转移方程很容易得到

对于每一个i-1,j的状态,要被每个使用不同钩子挂i个物品的状态继承,

初始化dp为0;

dp[0][7500]=1; 这里就是重点了,因为每一个dp[1]的情况都要从dp[0]开始继承,当放0件砝码时,天平必然处于平衡状态,所以dp[0]只有dp[0][7500]等于一.从程序实际意义上来讲,只有7500能满足减去7500力矩也能记录状态.

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int dp[21][15000];
int w[21];
int g[21];
int main()
{
    int m, n;
    scanf("%d%d", &m, &n);
    int i, j;
    int sum=0;
    int z=0;
    for(i=0; i<m; i++)
    {
        scanf("%d", &g[i]);
    }
    for(i=1; i<=n; i++)
    {
        scanf("%d", &w[i]);
    }
    memset(dp,0,sizeof(dp));
    dp[0][7500]=1;
    for(i=1; i<=n; i++)
    {
        for(j=0; j<=15000; j++)
        {
            if(dp[i-1][j]) //当i-1,j这个状态没有可能时,对它的继承也没用意义了.
            {
                for(int k=0; k<m; k++)
                dp[i][j+w[i]*g[k]]+=dp[i-1][j];
            }
        }
    }
    printf("%d\n", dp[n][7500]);
    return 0;
}