【回溯算法】最小重量机器设计问题

算法实现题 5-3 最小重量机器设计问题

问题描述：设某一机器由n个部件组成，每一种部件都可以从m个不同的供应商处购得。设wij是从供应商j处购来的部件i的重量，cij是相应的价格。
试设计一个算法，给出总价格不超过c的最小重量机器设计。
算法设计：对于给定的机器部件重量和机器部件价格，计算总价值不超过d的最小重量机器设计。

数据输入：第一行由3个正整数n，m，d。接下来的2n行，每行m个数。前n行是c，后n行是w。
结果输出:将计算的最小重量及每个部件的供应商输出。

输入：
3 3 4
1 2 3
3 2 1
2 2 2
1 2 3
3 2 1
2 2 2
  输出：
4
1 3 1

一个加长版的输入测试数据：

8 18 14
18 15 20 5 15 10 16 6 1 6 17 6 1 2 17 15 13 17
16 6 7 4 7 2 11 6 18 4 13 12 8 5 2 8 15 14
12 6 19 10 13 8 2 10 16 4 15 15 16 13 17 12 14 4
18 18 2 13 15 19 5 12 18 7 13 9 8 17 10 13 15 11
8 5 14 11 18 20 17 3 11 17 13 11 4 9 17 14 19 1
10 7 8 11 13 3 19 3 12 11 12 14 4 2 12 10 14 15
12 9 13 9 16 17 12 15 6 3 11 17 13 17 14 13 4 4
19 12 3 19 3 20 19 12 8 19 8 10 19 20 3 1 7 1
16 12 4 16 2 6 15 1 13 3 7 16 5 3 16 16 14 19
12 14 6 2 11 15 9 17 15 16 19 20 14 14 20 9 4 4
6 13 16 6 3 12 12 19 11 20 4 13 9 18 7 17 8 1
4 17 3 20 3 8 12 7 4 12 6 12 1 18 13 20 20 8
4 15 1 10 2 12 8 11 5 4 20 13 12 20 1 3 3 11
1 9 2 1 16 1 12 4 5 2 7 15 12 3 9 4 13 6
13 1 10 8 5 13 20 10 6 4 8 15 8 8 20 11 9 9
2 10 11 1 18 8 20 11 18 2 3 6 14 16 19 4 3 15

输出：
57
13 6 7 3 18 14 10 16
 

参考链接

https://blog.csdn.net/IOIO_/article/details/81093152

这种方法一般找到的都是第一个出现的解。有的时候第一个出现的解不一定是最优的，但是也要采取。

通过样例我们可以发现，其实1 3 2 或者 1 3 3 都是符合题目的要求的，而且最优情况也和 1 3 1 相同，但是由于他们按照代码的循环来找，找到之后就不再继续进行了，所以只有一个解。

最新版

#include<bits/stdc++.h>
using namespace std;

int n,m,d;//n个组件 m个供应商 总价格不超过d
int c[999][999],w[999][999];//c[i][j],w[i][j] 各表示从供应商j处购得部件i的价格/重量
int cw=0,cp=0;//记录当前最小的重量和价值
int bestw=999,bestp=999;//最小重量
int x[999],bestx[999];//保存当前重量下每个部件供应商的号码

void backtrack(int i) {
	if(i>n) {
		if(cp<=d&&cw<bestw) { //条件更优时，更新数据
			bestw=cw;
			bestp=cp;
			for(int j=1; j<=n; j++)
				bestx[j]=x[j];
		}

	} else {
		for(int j=1; j<=m; j++) {
			x[i]=j;//记录下当前部件是哪一个供应商
			cw=cw+w[i][j];
			cp=cp+c[i][j];
			if(cp<=d&&cw<bestw) {//剪枝
				//如果当前供应商的部件可以满足条件 直接进入下一个部件的判断
				backtrack(i+1);
			}
			//回溯过程
			cw=cw-w[i][j];
			cp=cp-c[i][j];
		}
	}
}

int main() {
	cin>>n>>m>>d;

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>c[i][j];

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>w[i][j];

	backtrack(1);
	cout<<bestw<<endl;
	for(int i=1; i<=n; i++)
		cout<<bestx[i]<<" ";
	return 0;
}
/*
输入：
3 3 4
1 2 3
3 2 1
2 2 2
1 2 3
3 2 1
2 2 2
输出：
4
1 3 1
*/

版本一 

#include<bits/stdc++.h>
using namespace std;

int n,m,d;//n个组件 m个供应商 总价格不超过d
int c[999][999],w[999][999];//c[i][j],w[i][j] 各表示从供应商j处购得部件i的价格/重量
int cw=0,cp=0;//记录当前最小的重量和价值 
int bestw=999;//最小重量 
int x[999],bestx[999];//保存当前重量下每个部件供应商的号码 

bool backtrack(int i) {
	if(i>n) {
		bestw=cw;
		for(int j=1; j<=n; j++)
			bestx[j]=x[j];
		return true;
	}
	bool find=false;
	if(bestw<=d)
		find=true;
	for(int j=1; j<=m; j++) {
		x[i]=j;//记录下当前部件是哪一个供应商 
		cw=cw+w[i][j];
		cp=cp+c[i][j];
		if(cp<=d&&cw<bestw) {//如果当前供应商的部件可以满足条件 直接进入下一个部件的判断 
			if(backtrack(i+1))
				find=true;
		}
		//回溯过程 
		cw=cw-w[i][j];
		cp=cp-c[i][j];
	}
	return find;
}

int main() {
	cin>>n>>m>>d;

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>c[i][j];

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>w[i][j];

	if(backtrack(1)) {
		cout<<bestw<<endl;
		for(int i=1; i<=n; i++)
			cout<<bestx[i]<<" ";
	}

	return 0;
}

版本二

#include<bits/stdc++.h>
using namespace std;

int n,m,d;//n个组件 m个供应商 总价格不超过d
int c[999][999],w[999][999];//c[i][j],w[i][j] 各表示从供应商j处购得部件i的价格/重量
int cw=0,cp=0;//记录当前最小的重量和价值
int bestw=999;//最小重量
int x[999],bestx[999];//保存当前重量下每个部件供应商的号码

void backtrack(int i) {
	if(i>n) {
		bestw=cw;
		for(int j=1; j<=n; j++)
			bestx[j]=x[j];
	}

	for(int j=1; j<=m; j++) {
		x[i]=j;//记录下当前部件是哪一个供应商
		cw=cw+w[i][j];
		cp=cp+c[i][j];
		if(cp<=d&&cw<bestw) {//如果当前供应商的部件可以满足条件 直接进入下一个部件的判断
			backtrack(i+1);
		}
		//回溯过程
		cw=cw-w[i][j];
		cp=cp-c[i][j];
	}
}

int main() {
	cin>>n>>m>>d;

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>c[i][j];

	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			cin>>w[i][j];

	backtrack(1);
	cout<<bestw<<endl;
	for(int i=1; i<=n; i++)
		cout<<bestx[i]<<" ";
	return 0;
}
/*
输入：
3 3 4
1 2 3
3 2 1
2 2 2
1 2 3
3 2 1
2 2 2
输出：
4
1 3 1
*/