回溯法解决0-1背包问题

给定n(n<=100)种物品和一个背包。物品i的重量是wi，价值为vi，背包的容量为C(C<=1000)。问:应如何选择装入背包中的物品，使得装入背包中物品的总价值最大? 在选择装入背包的物品时，对每种物品i只有两个选择：装入或不装入。不能将物品i装入多次，也不能只装入部分物品i。

输入格式:
共有n+1行输入： 第一行为n值和c值，表示n件物品和背包容量c； 接下来的n行，每行有两个数据，分别表示第i(1≤i≤n)件物品的重量和价值。

输出格式:
输出装入背包中物品的最大总价值。

输入样例:
在这里给出一组输入。例如：

5 10
2 6
2 3
6 5
5 4
4 6
输出样例:
在这里给出相应的输出。例如：

15

算法基本思想：
1.先将物品根据单位重量的价值由大到小排序；
2.依次装入满足条件的物品；
3.不满足条件时回退，求出上界函数，判断是否剪枝；

数据存储结构：vector
struct自定义结构体Bag

1.排序函数的重写

void sortPreValue(vector<struct Bag>bag){
	sort(bag.begin(),bag.end(),compare);
}

2.回溯函数

void backtrack(int i){
    int CId=i-1;
	double bound(int i);
	if(i>n){
		bestp=cp;
		return;
	}
	if(cw+bag[CId].wi<=c){
		cw+=bag[CId].wi;
		cp+=bag[CId].vi;
		put[CId]=1;
		backtrack(i+1);
		cw-=bag[CId].wi;
		cp-=bag[CId].vi;
	}
	if(bound(i+1)>bestp)//符合条件搜索右子树
		backtrack(i+1);
}

3.求上界函数

double bound(int i){
     int CId=i-1;
	int leftw=c-cw;
	double b=cp;
	while(i<=n && bag[CId].wi<=leftw){
		leftw-=bag[CId].wi;
		b+=bag[CId].vi;
		i++;
	}
	if(i<=n)
		b+=bag[CId].vi/bag[CId].wi*leftw;
	return b;
}

完整代码实现：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

struct Bag{
	int wi;
	int vi;
};

int compare(struct Bag &bag1,struct Bag &bag2){

return (bag1.vi/bag1.wi)>(bag2.vi/bag2.wi);
}

void sortPreValue(vector<struct Bag>bag){
	sort(bag.begin(),bag.end(),compare);
}

void backtrack(int i);

vector<struct Bag>bag;
struct Bag thing;

int n;//物品数量
double c;//背包容量
double cw = 0.0;//当前背包重量
double cp = 0.0;//当前背包中物品价值
double bestp = 0.0;//当前最优价值
int put[100];//设置是否装入

int main(){
int i;
cin>>n;
cin>>c;
for(i=1;i<=n;i++){
cin>>thing.wi;
cin>>thing.vi;
bag.push_back(thing);
}
sort(bag.begin(),bag.end(),compare); 
backtrack(1);
cout<<bestp;

return 0;
}

//回溯函数
void backtrack(int i){
    int CId=i-1;
	double bound(int i);
	if(i>n){
		bestp=cp;
		return;
	}
	if(cw+bag[CId].wi<=c){
		cw+=bag[CId].wi;
		cp+=bag[CId].vi;
		put[CId]=1;
		backtrack(i+1);
		cw-=bag[CId].wi;
		cp-=bag[CId].vi;
	}
	if(bound(i+1)>bestp)//符合条件搜索右子树
		backtrack(i+1);
}

//计算上界函数
double bound(int i){
     int CId=i-1;
	int leftw=c-cw;
	double b=cp;
	while(i<=n && bag[CId].wi<=leftw){
		leftw-=bag[CId].wi;
		b+=bag[CId].vi;
		i++;
	}
	if(i<=n)
		b+=bag[CId].vi/bag[CId].wi*leftw;
	return b;
}