BZOJ1433 [ZJOI2009]假期的宿舍 二分图匹配 匈牙利算法

原文链接http://www.cnblogs.com/zhouzhendong/p/8372785.html


题目传送门 - BZOJ1433


题解

  我们理一理题目。

  在校的学生,有自己的床,还可以睡朋友的床。

  离校的学生,不占床。

  外来的学生,只能睡朋友的床。

  然后就是一个裸的二分图匹配了。


代码

#include <cstring>
#include <cstdlib>
#include <cmath>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=55;
int T,n,g[N][N],a[N],b[N];
int match[N],vis[N];
int Match(int x){
	for (int i=1;i<=n;i++)
		if (!vis[i]&&g[x][i]){
			vis[i]=1;
			if (!match[i]||Match(match[i])){
				match[i]=x;
				return 1;
			}
		}
	return 0;
}
int hungray(){
	memset(match,0,sizeof match);
	for (int i=1;i<=n;i++)
		if (!b[i]){
			memset(vis,0,sizeof vis);
			if (!Match(i))
				return 0;
		}
	return 1;
}
int main(){
	scanf("%d",&T);
	while (T--){
		scanf("%d",&n);
		memset(g,0,sizeof g);
		for (int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		for (int i=1;i<=n;i++)
			scanf("%d",&b[i]);
		for (int i=1;i<=n;i++)
			for (int j=1;j<=n;j++){
				scanf("%d",&g[i][j]);
				if (!a[j])
					g[i][j]=0;
			}
		for (int i=1;i<=n;i++){
			if (a[i])
				g[i][i]=1;
			else
				b[i]=0;
		}
		puts(hungray()?"^_^":"T_T");
	}
	return 0;
}

  

posted @ 2018-01-28 22:54  -zhouzhendong-  阅读(135)  评论(0编辑  收藏