poj 1837 dp+01背包

题意,给出 n 个 挂钩的位置 ,- 表示在左边,+ 表示在右边,再给出m 个 砝码,现在要求 有多少种方法 能使 天平 平衡

思路,dp[i][j]表示挂 i 个砝码 力矩达到 j 的方法数,dp[i][j+hook[k]*val[i]]+=dp[i-1][j];也就是 dp[i][j]=sigma(dp[i-1][j-hook[k]*val[i]]);

#include<stdio.h>
#include<string.h>

const int maxn=25;
const int shift=7500;//15*20*25..力矩范围-7500~7500
int dp[maxn][shift<<1];
int hook[maxn],val[maxn];

int main(){
	int n,m;
	while(scanf("%d%d",&n,&m)==2){//

		for(int i=1;i<=n;i++)scanf("%d",hook+i);
		for(int i=1;i<=m;i++)scanf("%d",val+i);

		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++)dp[1][val[1]*hook[i]+shift]++;//初始化

		for(int i=2;i<=m;i++){
			for(int j=0;j<=2*shift;j++){
				for(int k=1;k<=n;k++){
					dp[i][j+hook[k]*val[i]]+=dp[i-1][j];
				}
			}
		}
		printf("%d\n",dp[m][shift]);
	}

}



你可能感兴趣的:(hook)