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

洛谷：矩阵游戏

洛谷题解就挺好

目的是使得最终（1，1）（2，2）…（n，n）都有一个点

可以看作为，最终状态需要每 i 行和 i 列都存在一个匹配

建图方式：对于 i 行 j 列的1点，建一条 i 连向 j 的边即可，最后跑一个二分图匹配，只有匹配数为 n 才能说明有解

可以证明交换行、交换列的操作不会影响匹配数

代码：

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define ul (u << 1)
#define ur (u << 1 | 1)
#define fx first
#define fy second
//#pragma GCC optimize(2)
//[博客地址](https://blog.csdn.net/weixin_51797626?t=1) 
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;

const int N = 210, M = 40010, MM = 3000010;
int INF = 0x3f3f3f3f, mod = 100003;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D;
int h[N], e[M], ne[M], idx;
int match[N << 1];
bool st[N << 1];

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

bool find(int x) { //标准匈牙利
	for (int i = h[x]; ~i; i = ne[i]) {
		int j = e[i];
		if (st[j])continue;
		st[j] = true;

		if (!match[j] || find(match[j])) {
			match[j] = x;
			return true;
		}
	}
	return false;
}

int main() {
	cinios;

	cin >> T;
	while (T--)
	{
		cin >> n;
		mem(h, -1);
		idx = 0;

		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++) {
				cin >> k;
				if (k)add(i, n + j);//行对列建边
				//虽然二分图是无向图问题，但只用建单向边即可，枚举其中一个集合
			}

		mem(match, 0);
		int mat = 0;
		for (int i = 1; i <= n; i++) {
			mem(st, 0);
			if (find(i))mat++;
		}

		if (mat == n)cout << "Yes";//匹配数必须为 n
		else cout << "No";
		cout << '\n';
	}

	return 0;
}