码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2018-04-01 23:06:53      阅读:247      评论:0      收藏:0      [点我收藏+]

标签:view   set   sed   blog   turn   ons   ace   log   span   

题意:

  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

 

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

标签:view   set   sed   blog   turn   ons   ace   log   span   

原文地址:https://www.cnblogs.com/Pneuis/p/8688618.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!