这道题目初看并非什么背包问题,很明显就是暴力+剪枝,暴力时间复杂度为O(20^20),铁定超时,剪枝也尝试过,还是没有发现比较好的剪枝方法,但还是贴一下暴力超时代码:
#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
#define Max 30
using namespace std;
int hooks[Max];
int weight[Max];
int c,g;
int Sum=0;
void dfs(int num,int Count){
if(num==g+1){
if(Count==0)
Sum++;
return ;
}
if(Count>0){ // 剪枝(极端法)
int temp=Count;
for(int i=num;i<=g;i++)
temp+=hooks[1]*weight[i];
if(temp>0)
return ;
}
else if(Count<0){ // 剪枝(极端法)
int temp=Count;
for(int i=num;i<=g;i++)
temp+=hooks[c]*weight[i];
if(temp<0)
return ;
}
for(int i=1;i<=c;i++)
dfs(num+1,weight[num]*hooks[i]+Count);
}
int main(){
scanf("%d%d",&c,&g);
for(int i=1;i<=c;i++)
scanf("%d",&hooks[i]);
for(int i=1;i<=g;i++)
scanf("%d",&weight[i]);
sort(hooks+1,hooks+c+1);
dfs(1,0);
printf("%d\n",Sum);
return 0;
}
即使上面已经有部分剪枝,但是时间复杂度仍然很高,而且剪枝效率不能保证。<pre class="cpp" name="code">这种情况下就要想到动态规划了(当暴力剪枝没法时,动态规划往往是另一条道路)
下面分析一下动态规划:
首先介绍一下度:即天平左右两端的力矩之和。
那么dp[i][j]定义如下:即将前i个物体放置于天平上所得度为j的不同种类的放置个数
接下来分析动态转移方程:
对于dp[i][j],第i个物体可以放置于天平上的k处(hooks[1]……hooks[c]),去掉已经放置了的第i个物体,那么剩下i-1个物体,且度为
j-hooks[k]*weight[i],那么将物体放置于天平上的k处时,则得到度为j的不同种类的放置个数为dp[i-1][j-hooks[k]*weight[i]],依次类推则可以推导出动态转移方程:
dp[i][j]=所有dp[i-1][j-hooks[k]*weight[i]]之和
当然题目规定:2<=c<=20 && (2<=g<=20) && 砝码最重为25 && 天平长度最大为15
则可以推导出度的范围为-20×25×15——20×25×15,也即-7500——7500,由于不能出现负度,故这里统一将所有度均加上7500,则度的范围为0——15000了。那么度为7500则对应天平平衡,时间复杂度为0(c*g*15000)完全可行。
下面是代码: 1912K+94MS
#include <stdlib.h>
#include <string.h>
#define Max 30
#define Maxx 15010
int hooks[Max];
int weight[Max];
int dp[Max][Maxx];
int c,g;
int main(){
scanf("%d%d",&c,&g);
for(int i=1;i<=c;i++)
scanf("%d",&hooks[i]);
for(int i=1;i<=g;i++)
scanf("%d",&weight[i]);
memset(dp,0,sizeof(dp));
dp[0][7500]=1; //将前0件物品放置于天平上,则度必须为7500,置其值为1;其它所有情况均为不合法情况,置其值为0
for(int i=1;i<=g;i++)
for(int j=0;j<=15000;j++) //均求出0———15000之间的所有dp,即使存在不合法的(将其值置为0)
for(int k=1;k<=c;k++){
if(j-weight[i]*hooks[k]>=0 && j-weight[i]*hooks[k]<=15000) //若不在0——15000之间则说明不合法,不在求和之中
dp[i][j]+=dp[i-1][j-weight[i]*hooks[k]];
}
printf("%d\n",dp[g][7500]); // 求出最后的个数,即将前i件物体放入天平,天平平衡时的不同放置个数
return 0;
}