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

給出$n\leq50$個人，然後一部分是在校學生，其中一部分要回家，另一部分留校。然後另外一部分是外來學生。然後每個人可以睡到和自己認識的人的牀位上。然後求問是否存在一種方案可以滿足所有的人都有睡覺的牀位。
對於所有的留校學生和外來學生都建邊連到自己可以睡的牀位。
然後做二分圖最大匹配，如果可以完全匹配，那麼滿足條件。

#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;
}