POJ 1719 Shooting Contest 二分匹配

题目链接：点击打开链接

给定n*m的矩阵

给定每列能匹配的2个行号

求一组解使得每列都匹配了一行，每行至少匹配一列

输出每列匹配的行号

思路：

先求个每行每列匹配的解，然后套上列空缺的

#include<iostream>
#include<set>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<vector>
using namespace std;
#define N 2020
int lef[N], pn;//lef[v]表示Y集的点v 当前连接的点 , pn为x点集的点数
bool T[N];     //T[u] 表示Y集 u 是否已连接X集
vector<int>G[N]; //匹配边  G[X集].push_back(Y集)  注意G 初始化

bool match(int x){ // x和Y集 匹配 返回x点是否匹配成功
	for(int i=0; i<G[x].size(); i++)
	{
		int v = G[x][i];
		if(!T[v])
		{
			T[v] = true;
			if(lef[v] == -1 || match( lef[v] ))   //match(lef[v]) : 原本连接v的X集点 lef[v] 能不能和别人连，如果能 则v这个点就空出来和x连
			{
				lef[v] = x;
				return true;
			}
		}
	}
	return false;
}

int solve(){
	int ans = 0;
	memset(lef, -1, sizeof(lef));
	for(int i = 1; i<= pn; i++)//X集匹配，X集点标号从 1-pn 匹配边是G[左点].size()
	{
		memset(T, 0, sizeof(T));
		if( match( i ) ) ans++;
	}
	return ans;
}

int n, m, a[N];
vector<int>P;
int main(){
    int t, i ,j;scanf("%d",&t);
    while(t--){
        scanf("%d %d",&n,&m);
        for(i = 1; i <= n; i++)G[i].clear();
        for(i = 1; i <= m; i++) {
            scanf("%d",&j);   G[j].push_back(i);
            scanf("%d",&j);   G[j].push_back(i);
            a[i] = j;
        }
        pn = n;
        int ans = solve();
        if(ans < n) {puts("NO");continue;}
        P.clear();
        for(i = 1; i <= m; i++) if(lef[i]!=-1)
            P.push_back(lef[i]);
            else P.push_back(a[i]);
        for(i = 0; i < P.size(); i++)
            printf("%d%c", P[i], i==P.size()-1?'\n':' ');
	}
	return 0;
}