bzoj 1059: [ZJOI2007]矩阵游戏（匈牙利算法）

1059: [ZJOI2007]矩阵游戏

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 4092   Solved: 1958
[ Submit][ Status][ Discuss]

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
【数据规模】
对于100%的数据，N ≤ 200

HINT

Source

[ Submit][ Status][ Discuss] 
HOME   Back

题解：匈牙利算法。

这道题有解的条件为存在n个行和列互相不相同的点。因为同一行或者同一列上的点无论怎么变化都不可能变得不同行或不同列。我们最后的要求是主对角线上都是黑点，所以要求选中的黑点不在同一行也不在同一列。这样在变化的时候才不会相互影响。我们可以根据条件建立二分图，一边表示行一边表示列，读入一个黑点就从他所属的行向所属的列连边，最后如果最大匹配数为n，则有解，否则无解。

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
#define N 430
using namespace std;
int tot,point[N*N],v[N*N],next[N*N],n,m;
int cur[N*N],belong[N*N];
void add(int x,int y)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
	tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}
bool find(int x,int k)
{
	for (int i=point[x];i;i=next[i]){
		int t=v[i];
		if (cur[v[i]]==k) continue;
		cur[v[i]]=k;
		if (!belong[v[i]]||find(belong[v[i]],k)) {
			belong[v[i]]=x;
			return true;
		}
	}
	return false;
}
int main()
{
	scanf("%d",&n);
	for(int t=1;t<=n;t++){
		scanf("%d",&m);
		tot=0;
		memset(next,0,sizeof(next));
		memset(point,0,sizeof(point));
		for (int i=1;i<=m;i++)
		 for (int j=1;j<=m;j++) {
		 	int x; scanf("%d",&x);
		 	if (x)  add(i,j+m);
		 }
		memset(belong,0,sizeof(belong));
		memset(cur,0,sizeof(cur));
		int ans=0;
		for(int i=1;i<=m;i++){
			if (find(i,i)) ans++;
			if (find(i+m,i+m)) ans++;
		}
		if (ans==2*m)  printf("Yes\n");
		else printf("No\n");
	}
}