1. 程式人生 > >【BZOJ】1433 [ZJOI2009]假期的宿舍 二分圖的最大匹配

【BZOJ】1433 [ZJOI2009]假期的宿舍 二分圖的最大匹配

題目的意思非常明顯,就是告訴你一些人之間的關係,一些床位可以容納一些人。問在校且不回家的人數加上外來人數是否能被所有床位容納。

這題需要注意每次求二分圖的最大匹配之前的初始化,我就是在這裡WA了好幾次……

之後只要貼上匈牙利演算法的模板就好了。

其實這題也可以用網路最大流來做,但是我覺得好像並不需要如此裝B吧……

附上AC程式碼:

#include <cstdio>
#include <cstring>
using namespace std;

int t,n,b[51],c[51],sum,x,a[51][51],ans,h[51];

bool so(int x){
	for (int i=1; i<=n; ++i)
		if (!b[i]&&a[x][i]){
			b[i]=1;
			if (!h[i]||so(h[i])){
				h[i]=x;
				return 1;
			}
		}
	return 0;
}

int main(void){
	scanf("%d",&t);
	while (t--){
		memset(a,0,sizeof a);
		memset(h,0,sizeof h);
		sum=ans=0;
		scanf("%d",&n);
		for (int i=1; i<=n; ++i) scanf("%d",&b[i]);
		for (int i=1; i<=n; ++i) {
			scanf("%d",&c[i]);
			if (!b[i]||(b[i]&&!c[i])) ++sum;
		}
		for (int i=1; i<=n; ++i){
			for (int j=1; j<=n; ++j){
				scanf("%d",&x);
				if (x&&(!b[i]||(b[i]&&!c[i]))&&b[j]) a[i][j]=1;
			}
			if (b[i]&&!c[i]) a[i][i]=1;
		}
		for (int i=1; i<=n; ++i){
			memset(b,0,sizeof b);
			if (so(i)) ++ans;
		}
		if (ans==sum) printf("^_^\n");
			else printf("T_T\n");
	}
	return 0;
}