POJ 1719 Shooting Contest（二分图匹配）

POJ 1719 Shooting Contest

题目链接

题意：给定一个矩阵，每列有两个白点。其它都是黑点，如今要求每列选一个白点，使得每一行至少包括一个白点被选中

思路：二分图匹配，白点的位置行列建边，然后跑匹配，假设匹配数不等于行数，就是是无解。然后输出方案的时候注意，假设有位置是没匹配的，说明这个位置是多余的，可是还是要随意选一个白点来输出

代码：

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int N = 1005;

int t, r, c;
vector<int> g[N];
int left[N], vis[N];

bool dfs(int u) {
	for (int i = 0; i < g[u].size(); i++) {
		int v = g[u][i];
		if (vis[v]) continue;
		vis[v] = 1;
		if (!left[v] || dfs(left[v])) {
			left[v] = u;
			return true;
		}
	}
	return false;
}

int save[N];

void hungary() {
	int ans = 0;
	memset(left, 0, sizeof(left));
	for (int i = 1; i <= r; i++) {
		memset(vis, 0, sizeof(vis));
		if (dfs(i)) ans++;
	}
	if (ans != r) {
		printf("NO\n");
		return;
	}
	for (int i = 1; i <= c; i++)
		printf("%d%c", left[i] ?

left[i] : save[i], i == c ? '\n' : ' '); } int main() { scanf("%d", &t); while (t--) { scanf("%d%d", &r, &c); for (int i = 1; i <= r; i++) g[i].clear(); int tmp; for (int i = 1; i <= c; i++) { for (int j = 0; j < 2; j++) { scanf("%d", &tmp); save[i] = tmp; g[tmp].push_back(i); } } hungary(); } return 0; }