P1129 [ZJOI2007]矩阵游戏 (匈牙利算法)

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

行交换操作：选择矩阵的任意两行，交换这两行（即交换对应格子的颜色）。
列交换操作：选择矩阵的任意两列，交换这两列（即交换对应格子的颜色）。
游戏的目标，即通过若干次操作，使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。

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

输入格式
本题单测试点内有多组数据。

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

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

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

输入输出样例
输入 #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$。

解题思路：将黑色格子所在的行和列连边, 跑一边最大匹配, 如果是完美匹配即可(所有点都是匹配点)
目标就是：每行都能匹配到一个不重复的列，而且这个行列坐标下是黑色格子。

匈牙利算法模板

int M, N;            //M, N分别表示左、右侧集合的元素数量
int Map[MAXM][MAXN]; //邻接矩阵存图
int p[MAXN];         //记录当前右侧元素所对应的左侧元素
bool vis[MAXN];      //记录右侧元素是否已被访问过
bool match(int i)
{
    for (int j = 1; j <= N; ++j)
        if (Map[i][j] && !vis[j]) //有边且未访问
        {
            vis[j] = true;                 //记录状态为访问过
            if (p[j] == 0 || match(p[j])) //如果暂无匹配，或者原来匹配的左侧元素可以找到新的匹配
            {
                p[j] = i;    //当前左侧元素成为当前右侧元素的新匹配
                return true; //返回匹配成功
            }
        }
    return false; //循环结束，仍未找到匹配，返回匹配失败
}
int Hungarian()
{
    int cnt = 0;
    for (int i = 1; i <= M; ++i)
    {
        memset(vis, 0, sizeof(vis)); //重置vis数组
        if (match(i))
            cnt++;
    }
    return cnt;
}

AC代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<time.h>
#include<vector>
using namespace std;
const int maxn = 205;
int Map[maxn][maxn],p[maxn],vis[maxn];
int n,t;
bool match(int i)
{
    for(int j=1; j<=n; ++j){
        if(Map[i][j]&&!vis[j]){
            vis[j] = 1;
            if(p[j]==0||match(p[j])){
                p[j] = i;
                return true;
            }
        }
    }
    return false;
}
int Hungarian()
{
    int cnt = 0;
    for(int i=1; i<=n; ++i){
        memset(vis,0,sizeof(vis));
        if(match(i))
            cnt++;
    }
    return cnt;
}
int main()
{
    //ios::sync_with_stdio(false);
    //cin.tie(0);
    //cout.tie(0);
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        memset(p,0,sizeof(p));
        for(int i=1; i<=n; ++i)
            for(int j=1; j<=n; ++j)
                scanf("%d",&Map[i][j]);
        if(Hungarian()==n)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}