【算法】求解最小机器重量设计问题回溯法（C++源码）

一、问题描述

设某一机器由n个部件组成，部件编号为1～n，
每一种部件都可以从m个供应商处购得，供应商编号为1～m。
设wij是从供应商j处购得的部件i的重量，cij是相应的价格。
对于给定的机器部件重量和机器部件价格，计算总价格不超过cost的最小重量机器设计，
可以在同一个供应商处购得多个部件。

二、输入描述

第一行输入3个整数n,m,cost，
接下来n行输入wij（每行m个整数）,
最后n行输入cij（每行n个整数）

三、输出描述

输出的第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

六、步骤描述

n个部件，m个供应商，不超过cost的总价，w[i][j]表示部件i从供应商j得到的重量，x[k]表示装每个零件对应的供应商的编号，深度优先搜索k的重量和价格，未超过则替换相加输出。

七、运行结果截图

八、源代码（C++）

#include <iostream>

using namespace std;

int m, cost;//n个部件，m个供应商，不能超过cost的开销
int w[100][100];//w[i][j]表示部件i从供应商j得到的重量
int c[100][100];
int x[100];//装每个零件对应的供应商的编号
int res[100];
int min_w=0x3f3f3f3f;

int dfs(int k, int weight, int pay)//每次搜第k个部件
{
	if(k==4)
	{
		if(weight<min_w)
		{
			min_w=weight;
			for(int i=0;i<m;i++)
			{
				res[i]=x[i];
			}
		}
		return 0;
	}
	else
    {
        for(int i=0;i<m;i++)
        {
            if(weight+w[k][i]<min_w && pay+c[k][i]<=cost)
            {
                x[k]=i;
                dfs(k+1,weight+w[k][i],pay+c[k][i]);
                x[k]=0;
            }
        }
    }
    return 0;
}

int main()
{
    int n;
	cin>>n>>m>>cost;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>w[i][j];
		}
	}
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>c[i][j];
		}
	}
	dfs(1,0,0);
	for(int i=0;i<m;i++)
	{
		cout<<res[i]+1<<" ";
	}
	cout<<endl;
	cout<<min_w+1<<endl;
	return 0;
}