最小重量机器设计问题

题目描述

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

输入描述

第一行有3 个正整数n，m和d。接下来的2n行，每行n个数。前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

思路：经典的回溯思想，深搜更新最优解
注意：需要利用前缀和剪枝，否者会超时
数据结构说明：

• final[i] = j 第i个零件来自的供应商j
• min_sc[i] 前i个零件的最小价格之和
• min_sw[i] 前i个零件的最小重量之和

---

Code:

#include<bits/stdc++.h>
using namespace std;
const int INF = 1e9 + 10;
const int N = 1e2 + 10;
int n,m,d;
int c[N][N],w[N][N],min_w=INF,final[N],ans[N];
int min_sc[N],min_sw[N],total_w,total_v;
void dfs(int dep) {if(dep==n+1) {if(total_w<min_w) { //更新min_w = total_w;for(int i=1; i<=n; i++) final[i] = ans[i];}return ;}for(int i=1; i<=m; i++) {int k = ans[dep];total_w += w[dep][i];total_v += c[dep][i];ans[dep] = i;if(total_v+min_sc[n]-min_sc[dep]<=d && total_w+min_sw[n]-min_sw[dep]<min_w) dfs(dep+1);total_v -= c[dep][i];total_w -= w[dep][i];ans[dep] = k;}
}
int main() {cin >> n >> m >> d;for(int i=1; i<=n*2; i++) {int k1 = INF;int k2 = INF;for(int j=1; j<=m; j++) {if(i<=n) {cin >> c[i][j];k1 = min(k1,c[i][j]);} else {cin >> w[i-n][j];k2 = min(k2,w[i-n][j]);}}if(i<=n) min_sc[i] = min_sc[i-1] + k1;else min_sw[i-n] = min_sw[i-n-1] + k2;}dfs(1);cout<<min_w;putchar('\n');for(int i=1; i<=n; i++ ) cout<<final[i]<<" ";return 0;
}
/**/