这个问题是典型的回溯法,比八皇后问题简化一点,重要的是其中一个部分:在深搜过程中要进行值的判断,来决定是否停止当前的搜索,这对以减少运行时间十分重要,一开始我没有考虑这个问题,就导致很多样例都超时了。
题目如下:
有 n 份工作要分配给 n 个人来完成,每个人完成一份。第 i 个人完成第 k 份工作所用的时间为一个正整数 tik,其中1 ≤ i, k ≤ n。试确定一个分配方案,使得完成这 n 份工作的时间总和最小。
输入
输入包含 n + 1 行。
第 1 行为一个正整数 n。
第 2 行到第 n + 1 行中每行都包含 n 个正整数,形成了一个 n × n 的矩阵。在该矩阵中,第 i 行第 k 列元素 tik 表示第 i 个人完成第 k 件工作所要用的时间。
输出
1 行,包含一个正整数,表示所有分配方案中最小的时间总和。
输入样例
5
9 2 9 1 9
1 9 8 9 6
9 9 9 9 1
8 8 1 8 4
9 1 7 8 9
输出样例
5
限制
1 ≤ n ≤ 15
1 ≤ tik ≤ 10^4
代码如下
#include <iostream>
using namespace std;
int N;
int people_work[15][15];
int vis[15]={0};
int cur[15];
int min_time;
int total=0;
void cal_min_time(int num){
if(num==N){
if(min_time==-1||total<min_time){
min_time=total;
}
}
else{
for(int i=0;i<N;i++){
if(!vis[i]){
total+=people_work[num][i];
if(min_time!=-1&&total>min_time){
total-=people_work[num][i];
continue;
}
vis[i]=1;
cal_min_time(num+1);
total-=people_work[num][i];
vis[i]=0;
}
}
}
}
int main()
{
cin>>N;
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
cin>>people_work[i][j];
}
}
min_time=-1;
cal_min_time(0);
cout<<min_time;
return 0;
}