题意:一个杠杆,中心有支点,两边有挂钩,挂钩位置由输入给定。有一些砝码,砝码重量由输入给定。一个挂钩下可以挂多个砝码,也可以什么都不挂,但每个砝码必须都用。问有多少种组合方式,可以使杠杆平衡。
分析:像这种组合方式很多,但状态空间却不大的,一般都可以交给DP。
dp[i][j]表示放置第i个物品后,总质量为j的可能组合的数量。
状态转移方程 dp[i+1][j]=sum{dp[i][j-wt[i]*h[k]](1<=k<=C)}
因为dp[i][j]中j可能为负数,在代码中给j加上一个偏移量temp。
代码:
#include<iostream>
#include<stdio.h>
#include<cstring>
using namespace std;
const int temp=4000;
int po[22],num[22],dp[22][2*temp];
int main()
{
int n,m,i,j,k;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(dp,0,sizeof(dp));
for(i=0;i<n;i++)
scanf("%d",&po[i]);
for(i=0;i<m;i++)
scanf("%d",&num[i]);
dp[0][temp]=1;
for(i=1;i<=m;i++)
for(j=temp-4000;j<temp+4000;j++)
for(k=0;k<n;k++)
dp[i][j]+=dp[i-1][j-po[k]*num[i-1]];
printf("%d\n",dp[m][temp]);
}
return 0;
}