POJ 1837 Balance(01背包问题)

Description
有一个天平,两个臂上有钩子,臂长15,给出所有钩子的位置,给出每个钩码的重量(各不相同),求必须使用所有钩码的前提下,有多少种平衡方法
Input
第一行为两个整数n和m分别表示钩子个数和钩码个数(2<=C,G<=20),第二行为钩子位置,第三行为钩码质量
Output
输出平衡方法种数
Sample Input
2 4
-2 3
3 4 5 8
Sample Output
2
Solution
a[i]表示钩子位置,b[i]表示钩子质量,dp[i][j]表示用前i个砝码到达力矩j的方法数
dp[i ][j+a[k]*b[i]]+=dp[i-1][j];
Code

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define maxn 21
#define maxv 15000
typedef long long ll;
ll n,m;
ll a[maxn],b[maxn];
ll dp[maxn][maxv];
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        dp[0][maxv/2]=1;//不挂钩码中心位置平衡 
        int i,j,k;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=1;i<=m;i++)
            scanf("%d",&b[i]);
        for(i=1;i<=m;i++)//m个钩码 
            for(j=1;j<=maxv;j++)//一个钩码只能用一次 
                if(dp[i-1][j])//判断前一状态是否存在 
                    for(k=1;k<=n;k++)//将这个钩码分别钩在n个钩子上 
                        dp[i][j+a[k]*b[i]]+=dp[i-1][j];
        printf("%d\n",dp[m][maxv/2]);//令中心位置平衡的种类数 
    }
    return 0;
}

你可能感兴趣的:(POJ 1837 Balance(01背包问题))