回溯法求解最小机器重量设计问题

求解最小机器重量设计问题
题目内容：

设某一机器由n个部件组成,部件编号为1_{n，每一种部件都可以从m个不同的供应商处购得，供应商编号为1}m。设wij是从供应商j处购得的部件i的重量,cij是相应的价格。对于给定的机器部件重量和机器部件价格，计算总价格不超过d的最小重量机器设计。（注意：输出结果中第一行最后没有空格。比如下面的输出样例中1 3 1后面没有空格。）

输入格式:

第1行输入3个正整数n,m和d。接下来n行输入wij（每行m个整数），最后n行输入cij（每行m个整数）,这里1≤n、m≤100。

输出格式：

输出的第1行包括n个整数，表示每个对应的供应商编号，第2行为对应的最小重量。

输入样例：

3 3 7

1 2 3

3 2 1

2 3 2

1 2 3

5 4 2

2 1 2

输出样例：

1 3 1

4

#include<cstdio>
#include<cstring>
using namespace std;

#define MAXSIZE 101
int n,m,d;
int w[MAXSIZE][MAXSIZE];//存储重量
int c[MAXSIZE][MAXSIZE];//存储价格
int x[2][MAXSIZE];//x[j]表示第j个商品所选取的店铺编号
int min_w=0x7fffffff;//最小的重量
int weight=0;
int price=0;

void get_w(int i){//选取第i个商品
	if(i>n){
	    if(weight<min_w){//商品都选择完毕，如果小于当前最小值，则替换
		    min_w=weight;
			for(int k=1;k<=n;k++){
				x[0][k]=x[1][k];//x[0]保存当前最小重量的商铺
			}
		}
		weight-=w[i-1][x[1][n]];//回溯，减去最后一件商品的相关重量和价格
		price-=c[i-1][x[1][n]];
	}
	else{
		int j;
		for(j=1;j<=m;j++){
			weight+=w[i][j];
			price+=c[i][j];
			if(price<=d){//价格小于d则递归求下一个商品
				x[1][i]=j;//第i个商品在第j个店铺选取
				get_w(i+1);
			}
			else{
			 	weight-=w[i][j];//否则回溯
			 	price-=c[i][j];
			}
		}
		if(j==m+1){
			weight-=w[i-1][x[1][i-1]];//循环结束再回溯一层
			price-=c[i-1][x[1][i-1]];
		}
	}
}

int main(){
	memset(w,0,sizeof(w));//置零
	memset(c,0,sizeof(c));
	memset(x,0,sizeof(x));
	scanf("%d%d%d",&n,&m,&d);//输入初始化
	
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&w[i][j]);
    for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&c[i][j]);
	get_w(1);
	
	for(int k=1;k<n;k++){
		printf("%d ",x[0][k]);
	}
		printf("%d\n",x[0][n]);
		printf("%d",min_w);
	return 0;
}