BZOJ 1059 矩阵游戏（二分图匹配，匈牙利算法）

题目链接：BZOJ 1059

通过分析，我们知道，行交换和列交换不会改变这一行的信息，这一行最多有一个1放在主对角线上。对于坐标为(i,j)的为黑色的棋子，我们只需将i行与j列连边，然后跑匈牙利算法即可。

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

int T,N;
int map[210][210],vis[210],belong[210];

bool find(int x){
	for(int i=1;i<=N;i++)
		if(map[x][i]&&!vis[i]){
			vis[i]=1;
			if(!belong[i]||find(belong[i])){
				belong[i]=x;
				return true;
			}
		}
	return false;
}

bool work(){
	for(int i=1;i<=N;i++){
		memset(vis,0,sizeof(vis));
		if(!find(i))return 0;
	}
	return 1;
}

int main(){
	scanf("%d",&T);
	while(T--){
		memset(belong,0,sizeof(belong));
		scanf("%d",&N);
		for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++){
				scanf("%d",&map[i][j]);
			}
		if(work())printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}