Description
有编号分别为1..n的n本书,准备分
给n个人,每个人阅读兴趣用一个二
维数组加以描述:
1:喜欢这本书
0:不喜欢这本书
Like[i][j] = 1 i 喜欢书j
Like[i][j] = 0 i 不喜欢书j
求解如何分书,让所有人都满意。
Input
输入有多组数据,每组数据第一行,包含一个的正整数n.
(0 < n <= 120).
接下来n行每行n个数。
Output
每组测试实例,输出一行,输出第i个分得的书的编号。每两个数据中间有个空格。
(默认从做到右依次是第1个人,第2个人…第n个人。) 若有多个答案,输出序列数小的那组。如 2 3 1 5 4和1 5 4 3 2 则输出1 5 4 3 2。因为(15432<23154)
Sample Input
2
0 1
1 1
Sample Output
2 1
分书问题,就是个深搜,不过写的时候要注意回溯。本来刚开始直接套裸的二分图匈牙利算法的模板的,结果错了……想想确实是错了,匈牙利是基于贪心思想,每一步都是当前步最优来匹配的,与这题的要求不符合,这题是要从第一个开始最优,第二个其次……依次向后的,所以dfs写法就不同了,这题主要是回溯这一步非常精妙。
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
using namespace std;
typedef long long ll;
const int N=1005;///两边的最大数量
int tu[N][N];
int to[N];///记录右边的点如果配对好了它来自哪里
bool use[N];///记录右边的点是否已经完成了配对
int n,m;///m,n分别表示两边的各自数量,n是左边,m是右边
bool dfs(int x)
{
if(x==n+1) return 1;
for(int i=1; i<=m; i++) ///m是右边,所以这里上界是m
if(!use[i]&&tu[x][i])
{
use[i]=1;
to[x]=i;
if(dfs(x+1))
return 1;
use[i]=0;///回溯!!!很重要
}
return 0;
}
void hungary()
{
memset(to,-1,sizeof(to));
memset(use,0,sizeof(use));
dfs(1);
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
scanf("%d",&tu[i][j]);
m=n;
hungary();
for(int i=1; i<=n; i++)
if(i!=n)printf("%d ",to[i]);
else printf("%d\n",to[i]);
}
return 0;
}