洛谷P2055 假期的宿舍 二分图匹配

给出$n\le 50$个人，然后一部分是在校学生，其中一部分要回家，另一部分留校。然后另外一部分是外来学生。然后每个人可以睡到和自己认识的人的床位上。然后求问是否存在一种方案可以满足所有的人都有睡觉的床位。
对于所有的留校学生和外来学生都建边连到自己可以睡的床位。
然后做二分图最大匹配，如果可以完全匹配，那么满足条件。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=53;
int a[N],b[N],c[N][N];
int mp[N][N]; 
bool used[N];
int link[N];
int n;
int dfs(int u) {
	for(int v=1;v<=n;v++) {
		if(mp[u][v]&&!used[v]) {
			used[v]=1;
			if(link[v]==-1||dfs(link[v])) {
				link[v]=u;
				return 1;
			}
		}
	}
	return 0;
}
int hungary() {
	int res=0;
	memset(link,-1,sizeof(link));
	for(int i=1;i<=n;i++) {
		memset(used,0,sizeof(used));
		if(dfs(i)) ++res;
	} 
	return res;
}
int main() {
	int T;
	scanf("%d",&T);
	while(T--) {
		scanf("%d",&n);
		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",&c[i][j]);
			}
		}
		memset(mp,0,sizeof(mp)); 
		for(int i=1;i<=n;i++) {
			if(a[i]&&(!b[i])) {
				mp[i][i]=1;
				for(int j=1;j<=n;j++) {
					if(c[i][j]&&a[j]) {
						mp[i][j]=1;
					}  
				}
			}
			else if(!a[i]) {
				for(int j=1;j<=n;j++) {
					if(c[i][j]&&a[j]) {
						mp[i][j]=1;
					}
				}
			}
		}
		int ans=hungary();
		int tot=0;
		for(int i=1;i<=n;i++) {
			if((a[i]&&(!b[i]))||!a[i]) ++tot; 
		}
		if(ans==tot) puts("^_^");
		else puts("T_T"); 
		
	}
	return 0;
}