7-4 工作分配问题 (20 分)
设有n件工作分配给n个人。将工作i分配给第j个人所需的费用为cij 。 设计一个算法,对于给定的工作费用,为每一个人都分配1 件不同的工作,并使总费用达到最小。
输入格式:
输入数据的第一行有1 个正整数n (1≤n≤20)。接下来的n行,每行n个数,表示工作费用。
输出格式:
将计算出的最小总费用输出到屏幕。
输入样例:
在这里给出一组输入。例如:
3
10 2 3
2 3 4
3 4 5
输出样例:
在这里给出相应的输出。例如:
9
知识点:本题用到了回溯法,(即dfs进行试探若不符合条件进行回溯)
思路:本题要找最小工资,即在每行每列找一个值使得总和是最小
自我纠正:之前dfs做习惯了每次dfs都想上下左右四个方向遍历,其实本题只需向下遍历,在用个book数组标记访问过的列,即可保证不同行不同列。(因为向下遍历肯定是不同行,再用book数组标记保证不同列)
源码:
#include<bits/stdc++.h>
using namespace std;
int num[101][101];
int book[101];
int mincost = 9999999;
int n;
void dfs(int i, int curcost) {
if (i > n) { // 寻找结束
if (curcost < mincost) //判断是否比最小值小
mincost = curcost;
return; // 无论执行不执行if语句都要return因为寻找已经结束
}
for (int j = 1;j <= n;j++) {
// 如何这个列还没有被访问并且当前工资加上num[i][j]
// 小于mincost进行循环,因为大于的话就没必要找了。
if (book[j] == 0 && curcost + num[i][j] < mincost) {
book[j] = 1;
curcost += num[i][j];
dfs(i + 1, curcost);
book[j] = 0; //进行回溯,因为要多次寻找确定最小值
curcost -= num[i][j]; // 回溯时要减掉之前的值
}
}
}
int main() {
cin >> n;
for (int i = 1;i <= n;i++) {
for (int j = 1;j <= n;j++) {
cin >> num[i][j];
}
}
dfs(1, 0); //进行深度搜素寻找最小工资
cout << mincost;
}