牛客KY59 神奇的口袋
题目描述
有一个神奇的口袋,总的容积是40,用这个口袋可以变出一些物品,这些物品的总体积必须是40。John现在有n个想要得到的物品,每个物品的体积分别是a1,a2……an。John可以从这些物品中选择一些,如果选出的物体的总体积是40,那么利用这个神奇的口袋,John就可以得到这些物品。现在的问题是,John有多少种不同的选择物品的方式。
输入
输入的第一行是正整数n (1 <= n <= 20),表示不同的物品的数目。接下来的n行,每行有一个1到40之间的正整数,分别给出a1,a2……an的值。
输出
输出不同的选择物品的方式的数目。
传送门:牛客KY59 神奇的口袋
解法一:递归
dfs(x,index),x代表当前剩余容量,index代表当前查询的物品编号。
如下代码:递归结束条件有两个,一是刚好目前容量为0,二是编号超过19。
很重要的一点是,为了避免重复,进入新的递归层不能从头开始,而是接续递归,保证搜索路径编号是递增的。
void dfs(int x,int index)
{
if(x==0){
ans++;
return;
}
if(index==20){
return;
}
for(int i=index;i<n;i++){
if(book[i]==0&&(x-v[i])>=0){
book[i]=1;
dfs((x-v[i]),i+1);
book[i]=0;
}
}
}
AC代码
#include "stdio.h"
int ans=0;
int n;
int v[20],book[20];
void dfs(int x,int index)
{
if(x==0){
ans++;
return;
}
if(index==20){
return;
}
for(int i=index;i<n;i++){
if(book[i]==0&&(x-v[i])>=0){
book[i]=1;
dfs((x-v[i]),i+1);
book[i]=0;
}
}
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&v[i]);
book[i]=0;
}
dfs(40,0);
printf("%d\n",ans);
return 0;
}
解法二 动态规划
先上代码,再分析。
#include "stdio.h"
int main()
{
int n;
int v[21];
int dp[25][45];//dp[i][j]代表前i个物品凑出j体积,共有多少种方法
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&v[i]);
dp[i][0]=1;//初始化边界
}
dp[0][0]=1;//初始化边界
for(int i=1;i<=n;i++){
for(int j=1;j<=40;j++){
//默认不能出现新的组合方式,那么则
dp[i][j]=dp[i-1][j];
//如果可以有新的组合方式
if(v[i]<=j){
dp[i][j]+=dp[i-1][j-v[i]];
}
}
}
printf("%d\n",dp[n][40]);
return 0;
}
关键点1:dp数组设置
dp[i][j]代表,前i个物品凑j的重量,有多少种方式。
这个我也理解的不是很透彻,感觉就是,数组的维度要和关键的变量相关,本题就是物品数量和容量。
但是,为什么是“前i个物品凑j的重量,有多少种方式”,而不是其他的XX方式,脑子里还不是很清晰,感觉还是要多做题,见多了可能会有更深刻的理解。
关键点2:边界设定
dp[x][0]=0,x∈[0,n]。
这个边界设定,决定着动态规划过程中的数量转移。
就是按照逻辑来设定,用前x个物品,容量为0的组合方式就一种,那就是啥都没有。
关键点3:转移方程
for(int i=1;i<=n;i++){
for(int j=1;j<=40;j++){
//默认不能出现新的组合方式,那么则
dp[i][j]=dp[i-1][j];
//如果可以有新的组合方式
if(v[i]<=j){
dp[i][j]+=dp[i-1][j-v[i]];
}
}
}
- i和j都要遍历到。
- 默认没有新的组合方式,那么
dp[i][j]=dp[i-1][j]
- 判断是否有新的组合方式,如果有,那么
dp[i][j]+=dp[i-1][j-v[i]]
(+=很重要,要理解) - 还有很重要的一点,需要理解。就是转移方程的右侧,一定是之前已经确定好的值,这样才能满足整个转移迭代的过程是正确的。这一点非常非常重要。