POJ 1837 Balance(01背包变型)

题意:

有一个杠杆,上面有c个钩子,有g件物品,全部放上去,问有多少种方式使天平平衡。

思路:

1. dp[i][j]代表第 i 件物品放到杆子上面之后,平衡度为 j 的方法数

2. 如果不加处理,j 可能会是负数,所以要加上一个 7500 的偏移,最后输出的结果也是 dp[n][7500]

3.  dp[i][v+w[i]*h[j]] += dp[i-1][v]; 是转移方程。初始化为 dp[0][7500] = 1. 因为什么都不放就是一种平衡方式。

 

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <algorithm>

using namespace std;



const int maxd = 15000;

const int maxn = 21;



int h[maxn], w[maxn];

int dp[maxn][maxd];



int main()

{

    int m, n;

    scanf("%d %d", &m, &n);



    for (int i = 0; i < m; ++i)

        scanf("%d", &h[i]);

    for (int i = 1; i <= n; ++i)

        scanf("%d", &w[i]);



    memset(dp, 0, sizeof(dp));

    dp[0][maxd>>1] = 1;



    for (int i = 1; i <= n; ++i)

        for (int v = 0; v <= maxd; ++v)

            if (dp[i-1][v])

                for (int j = 0; j < m; ++j)

                    dp[i][v+w[i]*h[j]] += dp[i-1][v];



    printf("%d\n", dp[n][maxd>>1]);

    return 0;

}

你可能感兴趣的:(poj)