BZOJ 1059 矩阵游戏 二分图匹配

题意:n*n(01)矩阵,操作:交换任意两行或者任意两列. 
n<=200.问是否能够通过若干次操作 使得矩阵左上到右下的主对角线变为全1.

原本在同一行或者同一列的点,经过操作后,还是在同一行或者同一列,同行同列上的两个点肯定不能同时作为主对角线元素
问题可以理解为 能否找到n个黑点,其任意两个不同行并且任意两个不同列.
二分图建图:左边为行 右边为列 若a[i][j]为黑点 则i->j连边. 
(i,j)边若在匹配被选中,则表示把(i,j)行换到(j,j).

若匹配数为n 则有解.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e2+20;
const ll mod=1e9+7;
int n,a[N][N],vis[N],mk[N]; 
bool find(int x)
{
	for(int i=1;i<=n;i++)
	{
		if(a[x][i]&& !vis[i])
		{
			vis[i]=1;
			if(!mk[i]||find(mk[i]))
			{
				mk[i]=x;
				return true;
			}
		}
	}
	return false;
}
bool check()
{
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		memset(vis,0,sizeof(vis));
		if(!find(i))
			return false;
	}
	return true;
}
int main()
{ 
	int T,x;
	cin>>T;
	while(T--)
	{
		scanf("%d",&n);
		memset(a,0,sizeof(a));
		memset(mk,0,sizeof(mk));
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				scanf("%d",&x);
				if(x)
					a[i][j]=1;
			}
		}
		if(check())
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}