POJ - 1469 COURSES (匈牙利算法入门题)

时间:2023-03-09 06:32:51
POJ - 1469 COURSES (匈牙利算法入门题)

题意:

  P门课程,N个学生。给出每门课程的选课学生,求是否可以给每门课程选出一个课代表。课代表必须是选了该课的学生且每个学生只能当一门课程的。

题解:

  匈牙利算法的入门题。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int maxn = ;
int t;
int k, s;
int flag;
int p, n;
int vis[maxn];
int match[maxn];
vector<int> g[maxn];
void init() {
flag = ;
memset(match, -, sizeof(match));
for(int i = ; i <= p+n; i++) g[i].clear();
}
bool dfs(int u) {
vis[u] = ;
for(int i = ; i < g[u].size(); i++) {
int v = g[u][i], w = match[v];
if(w< || !vis[w]&&dfs(w)) {
match[u] = v;
match[v] = u;
return true;
}
}
return false;
}
int hungarian() {
int res = ;
for(int u = ; u <= s+p; u++) {
if(match[u] < ) {
memset(vis, , sizeof(vis));
if(dfs(u)) res++;
}
}
return res;
}
int main() {
scanf("%d", &t);
while(t--) {
init();
scanf("%d%d", &p, &n);
for(int i = ; i <= p; i++) {
scanf("%d", &k);
if(!k) flag = ;
while(k--) {
scanf("%d", &s);
g[i].push_back(s+p);
g[s+p].push_back(i);
}
}
if(flag || n<p) {
puts("NO");
continue;
}
int ans = hungarian();
if(ans == p) puts("YES");
else puts("NO");
}
}