poj 1469 COURSES (匈牙利算法)

Posted by 111qqz on Thursday, May 26, 2016

TOC

poj 1469 题目链接

题意:p个课程,n个学生,给出每个课程对应选取该课程的学生编号,问能否选出p个学生,使得和课程一一对应。

思路:一眼二分图最大匹配。。。需要注意的是。。两个集合可能分别给出。。建图的最大点数是两个集合的最大数目之和。。。因为没注意这个细节RE了一次。。

/* ***********************************************
Author :111qqz
Created Time :2016年05月26日 星期四 23时53分39秒
File Name :code/poj/1469.cpp
************************************************ */

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#define fst first
#define sec second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ms(a,x) memset(a,x,sizeof(a))
typedef long long LL;
#define pi pair < int ,int >
#define MP make_pair

using namespace std;
const double eps = 1E-8;
const int dx4[4]={1,0,0,-1};
const int dy4[4]={0,-1,1,0};
const int inf = 0x3f3f3f3f;
const int N=30003;
int n,p;
int head[405];
int cnt ;
int link[405];
bool vis[405];
struct Edge
{
    int v;
    int nxt;
}edge[N];

void addedge( int u,int v)
{
    edge[cnt].v = v;
    edge[cnt].nxt = head[u];
    head[u] = cnt;
    cnt++;
}


bool dfs( int u)
{
  //  cout<<"u:"<<u<<endl;
    for ( int i = head[u] ; i!=-1 ; i = edge[i].nxt)
    {
    int v = edge[i].v;
    if (vis[v]) continue;
    vis[v] = true;

    if (link[v]==-1||dfs(link[v]))
    {
        link[v] = u;
        return true;
    }
    }
    return false;
}
int hungary()
{
    int res = 0 ;
    ms(link,-1);

    for ( int i = 1 ; i <= p ; i++)
    {
    ms(vis,false);
    if (dfs(i))  res++;
    }
    
    return res;

}
int main()
{
    #ifndef  ONLINE_JUDGE 
    freopen("code/in.txt","r",stdin);
  #endif
    int T;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&p,&n); 
        cnt  = 0;
        ms(head,-1);
        ms(edge,0);
        for ( int i = 1 ; i <= p ; i++)
        {
        int x;
        scanf("%d",&x);
        while (x--)
        {
            int y;
            scanf("%d",&y);
            y+=p;
            addedge(i,y);
        }
        }
        int ans = hungary();
	    
        if (ans==p) puts("YES");
        else puts("NO");
    }


  #ifndef ONLINE_JUDGE  
  fclose(stdin);
  #endif
    return 0;
}

「真诚赞赏,手留余香」

111qqz的小窝

真诚赞赏,手留余香

使用微信扫描二维码完成支付