poj 1469 COURSES 解题报告

题目链接：http://poj.org/problem?id=1469

题目意思：有 N 个人，P个课程，每一个课程有一些学生参加（0个、1个或多个参加）。问 能否使得 P 个课程 恰好与 P 个学生相匹配。

    受之前第一道匹配题（POJ 1274 The Perfect Stall）的影响，没有仔细体会是从哪个点匹配到哪个点，这将导致Hungary() 和 dfs() 中 for 循环的约束条件，究竟是遍历课程数 P 还是 学生数N，不要搞混！其实从存储的map[i][j] 可以 知道 怎样遍历的啦 ^_^

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 300 + 5;
bool vis[maxn];
int map[maxn][maxn], match[maxn];
int N, P, cnt;

int dfs(int x)
{
    for (int i = 1; i <= N; i++)   // 为p个课程找匹配,N个学生是候选人
    {
        if (!vis[i] && map[x][i])
        {
            vis[i] = 1;
            if (!match[i] || dfs(match[i]))
            {
                match[i] = x;
      //          printf("match[%d] = %d
", i, match[i]);
                return 1;
            }
        }
    }
    return 0;
}

void Hungary()
{
    cnt = 0;
    for (int i = 1; i <= P; i++)  // 不是N，要区分清楚！！
    {
        memset(vis, 0, sizeof(vis));
        cnt += dfs(i);
    }
}

int main()
{
    int T, stu_num, stu_id;
    while (scanf("%d", &T) != EOF)
    {
        while (T--)
        {
            memset(match, 0, sizeof(match));
            memset(map, 0, sizeof(map));

            scanf("%d%d", &P, &N);
            for (int i = 1; i <= P; i++)
            {
                scanf("%d", &stu_num);
                for (int j = 0; j < stu_num; j++)
                {
                    scanf("%d", &stu_id);
                    map[i][stu_id] = 1;
                }
            }
            Hungary();
            printf("%s
", cnt == P ? "YES" : "NO");
        }
    }
    return 0;
}