洛谷 P1129 [ZJOI2007]矩阵游戏（二分图匹配）

题目描述

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

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

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

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

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

输入输出格式

输入格式：

 

第一行包含一个整数TT，表示数据的组数。

接下来包含TT组数据，每组数据第一行为一个整数NN，表示方阵的大小；接下来NN行为一个N \times NN×N的0101矩阵（00表示白色，11表示黑色）。

 

输出格式：

 

包含TT行。对于每一组数据，如果该关卡有解，输出一行YesYes；否则输出一行NoNo。

 

输入输出样例

输入样例#1： 

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

输出样例#1： 

No
Yes

说明

对于20\%20%的数据，N ≤ 7N≤7

对于50\%50%的数据，N ≤ 50N≤50

对于100\%100%的数据，N ≤ 200N≤200

解题思路

行和列二分，若坐标(x,y)为1，连接x和y，然后求最大匹配，若为n则为“Yes”，否则为"No"。如果交换了x和x+1行，就相当于在二分图中把点x和x+1交换了，并不会影响匹配数。对于目标的结果（对角线上全有），需要有n个匹配，才能满足。

代码如下

#include <iostream>
#include <cstring>
#define maxn 205
using namespace std;
bool g[maxn][maxn];
int pre[maxn];
int n;
bool vis[maxn];
bool dfs(int x)
{
    for(int i = 1; i <= n; i ++){
        if(g[x][i] && !vis[i]){
            vis[i] = true;
            if(!pre[i] || dfs(pre[i])){
                pre[i] = x;
                return true;
            }
        }
    } 
    return false;
}
bool max_match()
{
    memset(pre, 0, sizeof(pre));	
    for(int i = 1; i <= n; i ++){
        memset(vis, 0, sizeof(vis));
        if(!dfs(i))
            return false;
    }
    return true;
}
int main()
{
    int T;
    cin >> T;
    while(T --){		
        cin >> n;
        for(int i = 1; i <= n; i ++){
            for(int j = 1; j <= n; j ++)
                cin >> g[i][j];
        }
        if(max_match())
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }
    return 0;
}