问题描述:
有n个任务需要分配给n个人执行,每个任务只能分配给一个人,每个人只能执行一个任务,第i个人执行第j个任务的成本是c【i】【j】。求出总成本最小的一种方案。
求解:
任务的选择依旧可以表示成一棵n叉树,每次父节点向孩子结点的延申,就是当前i号人员选择第j个任务。
设计递归,当i递归到==n,说明所有人都选择了一个互相之间各不相同的任务,这便是出口,也是一次结果,用temp【】来记录本次人员的选择,用temp_cost来记录本次结果,misson【】数组保证每个任务只能被选择一次。
设计剪枝,我们发现当走到某个非叶子节点后的temp_cost已经大于之前的结果了,我们可以把它剪掉。
代码:
#include<stdio.h>
#include<iostream>
using namespace std;
#define max 10
int min_cost = 999;
void dis(int* x, int n) {
for (int i = 0; i < n; i++)
cout << x[i] << " ";
cout << endl;
}
bool check_mission(int* mission, int n) {
for (int i = 0; i < n; i++) {
if (mission[i] == 0)
return false;
}
return true;
}
void dfs(int c[max][max], int* mission, int* temp, int* x, int temp_cost,int n, int i) {
if (i == n) {
if (temp_cost < min_cost) {
min_cost = temp_cost;
for (int j = 0; j < n; j++)
x[j] = temp[j];
}
}
else {
for (int j = 0; j < n; j++) {
if (mission[j] == 0) {
if (temp_cost+c[i][j] > min_cost)
continue;
temp[i] = j;
mission[j] = 1;
dfs(c, mission, temp, x, temp_cost + c[i][j],n, i + 1);
mission[j] = 0;
}
}
}
}
int main() {
int c[max][max] = { {9,2,7,8},{6,4,3,7},{5,8,1,8},{7,6,9,4} };
int mission[max];
int x[max];
int temp[max];
int n = 4;//一共四人个人,一共四种任务
memset(temp, -1, sizeof(temp));
memset(x, -1, sizeof(x));
memset(mission, 0, sizeof(mission));
int temp_cost = 0;
dfs(c, mission, temp, x, temp_cost,n, 0);
dis(x, n);
return 0;
}