POJ - 1469 COURSES （匈牙利算法入门题）

Posted 2020-10-29 Pneuis

题意：

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

题解：

　　匈牙利算法的入门题。

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

View Code