最优装载问题-回溯算法

题目描述

同算法设计与分析课本，找出一个最优装载方式使第一个集装箱尽可能装满

代码介绍

#include<bits/stdc++.h>
using namespace std;
const int N=520;
int r;//未被选择的集装箱的总重量
int bestw;//当前最优装载量
int cw;//当前装载量
int x[N];//保存集装箱数组
int w[N];//保存各个集装箱重量
int n;//集装箱的个数
int c1,c2;//两艘船各自的装载量
void dfs(int u){
	if(u>n){
		if(cw>bestw){
			bestw=cw;
		}
		return ;
	}//到达叶节点，判断是否需要更新
	r-=w[u];//将该集装箱从剩余的集装箱中去掉
	if(cw+w[u]<c1){//搜寻左子树x[i]=1;
		x[u]=1;
		cw+=w[u];
		dfs(u+1);//搜寻下一层
		x[u]=0;
		cw-=w[u];//进行回溯
	}//将这个集装箱加入集合的情况
	if(r+cw>bestw){//搜寻右子树x[i]=0,同时若剩下的总重量比当前bestw还小，则一定不满足条件
		x[u]=0;
		dfs(u+1);
	}
	//以上两段可以用for(int i=0;i<=1;i++) x[u]=i;  界定条件 
	r+=w[u];;//进行回溯
}
int main(){
	cin>>c1>>c2;
	for(int i=1;i<=n;i++){
		cin>>w[i];
		r+=w[i];
	} 
	dfs(1);//从第一个集装箱开始遍历
	cout<<bestw<<endl;//输出第一个集装箱的最优装载量
	return 0;
}