深度优先搜索(DFS)之01背包问题:
题目描述:
一共有N件物品,每件物品的重量为w[i],价值为v[i]。在总重量不超过背包承载上限v的情况下,能够装入背包的最大价值是多少?(1<= n<= 20)
- 首先思路就是每个物品可以选可以不选,在总重量小于规定重量的情况下得出最大价值。
及n件物品,选的话加上这个物品的重量和价值,不选的话直接数组下标加一,边界就是数组下标==数量n,如果再结合剪枝的思想,可以在“选”这个分支上加上判断条件重量是否会超标,如果不超标则比较一下现在背包里的价格和记录最大价值的变量,如果大于则更新最大价值。
dfs部分代码:
void dfs(int index, int sumw, int sumv){
if(index == n){
if(sumw<=c&&sumv>maxv) maxv = sumv;
return;
}
dfs(index+1, sumw, sumv);
if(sumw+w[index]<= c){
if(sumv+v[index]> maxv) maxv = sumv+v[index+1];
dfs(index+1, sumw+w[index], sumv+v[index]);
}
}
总代码
#include<iostream>
using namespace std;
const int maxn = 30;
int n, c, maxv = 0;
int w[maxn], v[maxn];
void dfs(int index, int sumw, int sumv){
if(index == n){
if(sumw<=c&&sumv>maxv) maxv = sumv;
return;
}
dfs(index+1, sumw, sumv);
if(sumw+w[index]<= c){
if(sumv+v[index]> maxv) maxv = sumv+v[index+1];
dfs(index+1, sumw+w[index], sumv+v[index]);
}
}
int main(){
cin>>n>>c;
int i;
for(i = 0; i< n; i++){
cin>>w[i];
}
for(i = 0; i< n; i++){
cin>>v[i];
}
dfs(0, 0, 0);
cout<<maxv<<endl;
return 0;
}