POJ 1837.Balance

题目:http://poj.org/problem?id=1837

AC代码(C++):

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <vector>
#include <queue>
#include <math.h>
#include <string>
#include <string.h>
#include <bitset>

#define INF 0xfffffff
#define MAXN 100105

using namespace std;

int c,g;
int p[25],w[25];
int dp[25][15005];

int main(){
	cin>>c>>g;
	for(int i = 0; i < c; i++)cin>>p[i];
	for(int i = 0; i < g; i++)cin>>w[i];
	memset(dp,0,sizeof(dp));
	dp[0][7500] = 1;
	for(int i = 1; i <= g; i++){
		for(int j = 0; j < 15005; j++){
			for(int k = 0; k < c; k++){
				if(j+p[k]*w[i-1]>=15005||j+p[k]*w[i-1]<0)continue;
				dp[i][j+p[k]*w[i-1]]+=dp[i-1][j];
			}
		}
	}
	cout<<dp[g][7500]<<endl;
}

总结: 01背包问题, dp[i][j]为放入第i个砝码后平衡位置在j的情况数. 因为平衡位置j作为下标不能取负数, 所以移15*20*25位后平衡位置在7500. 初始时天平在第0行的平衡位置7500, 即dp[0][7500] = 1. 之后一个一个砝码放, 每个砝码在不同的位置都要放, 直到放完所有砝码, 则dp[g][7500]就是答案.