洛谷 P1129 矩阵游戏 题解

题目链接：P1129 矩阵游戏

题目描述

小 Q 是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏――矩阵游戏。矩阵游戏在一个 $n\times 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\le 7$。
• 对于 $50\%$ 的数据，保证 $n\le 50$。
• 对于 $100\%$ 的数据，保证 $1\le n\le 200$，$1\le T\le 20$。

题目思路

由题意得，我们可以对任意两行或两列进行交换操作。初看似乎没有头绪，那先来分析一组简单数据。

1
0 0 1
1 0 0
0 1 0

我们考虑将每一行的行数与该行黑色方块所在的列数连一条边。
然后可以将它进行如下图的变化得到答案
我们可以看出，右图中的每一行列都至少有一个符合的匹配（显然）。
进而我们思考，是否是每一行列至少存在一组合法匹配的图形呢？
答案是的
因为任意两行或者任意两列之间的交换都不会破坏他们的平衡性。

也就是说，任意2个黑色方块，如果它们初始状态时不在同一行（列），那么无论如何交换，它们都不会在同一行（列）。
所以我们可以得到解法 匈牙利二分图匹配解法。

匈牙利二分图匹配解法

经过上述分析，交换操作并不会影响图的最大匹配，我们只需判断每一行是否都可以合法匹配即可。
那么显然是一道板子似的二分图最大匹配。

代码

/*************************************************
Note:
*************************************************/
#include <queue>
#include <stack>
#include <set>
#include <stdio.h>
#include <iostream>
#include <vector>
#include <iomanip>
#include <string.h>
#include <algorithm>
#include <cmath>
#include <cstring>
#define ll long long
#define ull unsigned long long
using namespace std;
const int N = 1e4 + 10;
const int INF = 0x3f3f3f3f;
inline int read()
{
    int s = 0, w = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9')
    {
        if (ch == '-')
            w = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9')
        s = s * 10 + ch - '0', ch = getchar();
    return s * w;
}
int head[N * N], to[N * N], nt[N * N], tot = 1, ans = 0;
int match[N], visit[N], n, T;
void add(int x, int y)
{
    tot++, to[tot] = y, nt[tot] = head[x], head[x] = tot;
}

bool dfs(int x)
{ 
    for (int i = head[x], y; i; i = nt[i])
        if (!visit[y = to[i]])
        {
            visit[y] = 1;
            if (!match[y] || dfs(match[y]))
            {
                match[y] = x;
                return true;
            }
        }
    return false;
}

void clear()
{ 
    T--;
    ans = 0;
    for (int i = 1; i <= tot; i++) //这种方式可以避免memset()浪费过多时间
        to[i] = head[i] = nt[i] = 0;
    tot = 1;
    for (int i = 1; i <= 2 * n; i++)
    {
        match[i] = 0;
    }
}

int main()
{
    T = read();
    while (T != 0)
    {
        n = read();
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
            { 
                int m1 = read();
                if (m1 == 1)
                    add(i, j + n);
            }
        for (int i = 1; i <= n; i++)
        {
            memset(visit, 0, sizeof(visit));
            if (dfs(i))
                ans++;
        }
        if (ans >= n)
            printf("Yes\n"); 
        else
            printf("No\n");
        clear();
    }
    return 0;

提示：$clear()$函数中 清零时用了多少就清零多少，有效节约因$memset()$清零空数组而浪费的时间