C++矩阵游戏

题目描述

小 Q 是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏――矩阵游戏。矩阵游戏在一个 n*n 黑白方阵进行（如同国际象棋一般，只是颜色是随意的）。每次可以对该矩阵进行两种操作：

• 行交换操作：选择矩阵的任意两行，交换这两行（即交换对应格子的颜色）。

• 列交换操作：选择矩阵的任意两列，交换这两列（即交换对应格子的颜色）。

游戏的目标，即通过若干次操作，使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。

对于某些关卡，小 Q 百思不得其解，以致他开始怀疑这些关卡是不是根本就是无解的！于是小 Q 决定写一个程序来判断这些关卡是否有解。

输入格式

第一行包含一个整数 T，表示数据的组数，对于每组数据，输入格式如下：

第一行为一个整数，代表方阵的大小 n。 接下来 n 行，每行 n 个非零即一的整数，代表该方阵。其中 0 表示白色，1 表示黑色。

输出格式

对于每组数据，输出一行一个字符串，若关卡有解则输出 Yes，否则输出 No。

样例 #1

样例输入 #1

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

样例输出 #1

No
Yes

提示

数据规模与约定

• 对于 20% 的数据，保证 n《=7。

• 对于 50% 的数据，保证 n<=50。

• 对于 100% 的数据，保证 1<=n<=200,1<=T<=20。

解题思路

将黑色格子所在的行和列的边连接起来，使得一边最大匹配，若为完美匹配，则达成条件(所有点都为匹配点)

在完美匹配的情况下：

假如 列1 -> 行5；列4 ->行1(->表示匹配)

则可以将行5和行1进行位置交换，这种交换方式不会受到影响

交换后匹配变为：

列1 -> 行1；列4 -> 行5

解题代码如下

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N = 40005;
int h[N], ne[N * 2], to[N * 2];
int tot, T, n;

void add(int x,int y) {
	ne[++tot] = h[x];
	h[x] = tot, to[tot] = y;
}

int read(void) {
	int x = 0;
	char c = getchar();
	while (!isdigit(c)) c = getchar();
	while (isdigit(c)) {
		x = (x << 3) + (x << 1) + c - '0';
		c = getchar();
	}
	return x;
}

int tim, vis[N], match[N];

int dfs(int x) {
	for (int i = h[x]; i; i = ne[i]) {
		int y = to[i];
		if (vis[y] == tim) continue;
		vis[y] = tim;
		if (!match[y] || dfs(match[y])) {
			match[y] = x; return 1;
		}
	}
	return 0;
}

void pre(void) {
	memset(h, 0, sizeof(h));
	memset(ne, 0, sizeof(ne));
	memset(to, 0, sizeof(to));
	memset(match, 0, sizeof(match));
	tot = 0;
}

int main() {
	T = read();
	while (T--) {
		pre();
		n = read();
		for (int i = 1;i <= n; i++) 
		for (int j = 1;j <= n; j++) {
			int x = read();
			if (x) {
				add(i, j + n); add(j + n, i);
			}
		}
		int ans = 0;
		for (int i = 1;i <= n; i++) 
		tim++, ans += dfs(i);
		if (ans == n) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}