input.txt
3 3 4
3 2 1
1 2 3
1 2 2
1 2 3
3 2 1
2 2 2
#include <iostream>
#include <fstream>
using namespace std;
int main(){
//读取文件
ifstream in("input.txt", ios::in);
if (!in.is_open()){
cout << "Error opening file";
exit(1);
}
int n = in.get() - '0';
in.get();
int m = in.get() - '0';
in.get();
int d = in.get() - '0';
in.get();
int** c = new int*[n];
for (int i = 0; i < n; i++){
c[i] = new int[m];
}
int** w = new int*[n];
for (int i = 0; i < n; i++){
w[i] = new int[m];
}
for (int i = 0; i < 2 * n; i++){
if (i < n){
for (int j = 0; j < m; j++){
c[i][j] = in.get() - '0';
in.get();
}
}
else if ((i < 2 * n) && (i >= n)){
for (int j = 0; j < m; j++){
w[i-3][j] = in.get() - '0';
in.get();
}
}
}
cout << n << " " << m << " " << d << endl;
for (int i = 0; i < n; i++){
cout << c[i][0] << " " << c[i][1] << " " << c[i][2] << endl;
cout << w[i][0] << " " << w[i][1] << " " << w[i][2] << endl;
}
//读取文件完毕
//数据处理
//已初始化数据n,m,d,c[][],w[][]
bool flag = true;
int ci = 0;//当前层
int* bestx = new int[n];//记录最佳序列
int* x = new int[n];//记录当前序列
int bestw = INT_MAX;//最小重量
int nc = 0;//当前价值
int nw = 0;//当前重量
int j = 0;//第一个
int* min = new int[n];
min[n - 1] = 0;
int r = 0;
for (int i = n-1; i >0; i--){
r = w[i][0];
for (int j = 1; j < m; j++){
if (w[i][j] < r){
r = w[i][j];
}
}
min[i-1] = r+min[i];
}
while (flag){
while ((ci < n) && (nc < d)){//向左走(上界)
nc += c[ci][j];
nw += w[ci][j];
x[ci] = j;
ci++;
if (ci != n){//到叶子就不跑左子树
j = 0;
}
}
if ((ci == n)&&(nc<=d)){//到叶子
if (nw < bestw){
bestw = nw;
for (int z = 0; z < n; z++){
bestx[z] = x[z];
}
}
}
//减去多余
ci--;
nc -= c[ci][x[ci]];
nw -= w[ci][x[ci]];
j = x[ci];
do{//移动当前层的下一个节点
j++;
} while ((j < m) && (nc + c[ci][j]>d) );//判断下一个节点是否可用
while ((bestw != 0) && (nw + min[ci] + w[ci][j] >= bestw)&&(j<m)){//判断有没有必要移动到下一个节点(下界)
j++;
}
while (j == m){//当该层跑完回溯上层直到有一层还没遍历完的
ci--;
if (ci == -1){
flag = false;
break;
}
nc -= c[ci][x[ci]];
nw -= w[ci][x[ci]];
j = x[ci] + 1;
}
}
cout << "最小重量" << bestw << endl;
for (int i = 0; i < n; i++){
cout << " " << 1+bestx[i];
}
cout << endl;
return 0;
}