bzoj 1059: [ZJOI2007]矩阵游戏 二分图匹配

1059: [ZJOI2007]矩阵游戏

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1891 Solved: 919
[Submit][Status]

Description

小Q是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏——矩阵游戏。矩阵游戏在一个N*N黑白方阵进行（如同国际象棋一般，只是颜色是随意的）。每次可以对该矩阵进行两种操作：行交换操作：选择矩阵的任意两行，交换这两行（即交换对应格子的颜色）列交换操作：选择矩阵的任意行列，交换这两列（即交换对应格子的颜色）游戏的目标，即通过若干次操作，使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。对于某些关卡，小Q百思不得其解，以致他开始怀疑这些关卡是不是根本就是无解的！！于是小Q决定写一个程序来判断这些关卡是否有解。

Input

第一行包含一个整数T，表示数据的组数。接下来包含T组数据，每组数据第一行为一个整数N，表示方阵的大小；接下来N行为一个N*N的01矩阵（0表示白色，1表示黑色）。

Output

输出文件应包含T行。对于每一组数据，如果该关卡有解，输出一行Yes；否则输出一行No。

Sample Input

2
2
0 0
0 1
3
0 0 1
0 1 0
1 0 0

Sample Output

No
Yes
【数据规模】
对于20%的数据，N ≤ 7
对于50%的数据，N ≤ 50
对于100%的数据，N ≤ 200

HINT

匈牙利算法又有点忘了，这类棋盘转化问题不一定要到目标棋盘，只要发现规律即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 300
vector<int> V[MAXN];
int vis[MAXN];
int pat[MAXN];
int find(int x)
{
int i;
for (i=0;i<V[x].size();i++)
{
if (vis[V[x][i]])continue;
vis[V[x][i]]=true;
if (pat[V[x][i]]==-1||find(pat[V[x][i]]))
{
pat[V[x][i]]=x;
return 1;
}
}
return 0;
}
int main()
{
freopen("input.txt","r",stdin);
//freopen("output.txt","w",stdout);
int nn,n,m;
int i,j,k;
int x;
int tot;
scanf("%d",&nn);
while (nn--)
{
scanf("%d",&n);
for (i=0;i<n;i++)
{
for (j=0;j<n;j++)
{
scanf("%d",&x);
if (x)V[i].push_back(j);
}
}
tot=0;
memset(pat,-1,sizeof(pat));
for (i=0;i<n;i++)
{
memset(vis,0,sizeof(vis));
tot+=find(i);
}
if (tot==n)
{
printf("Yes\n");
}else
{
printf("No\n");
}
for(i=0;i<n;i++)V[i].clear();
}
}