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 =  + ;
 bool vis[maxn];
 int map[maxn][maxn], match[maxn];
 int N, P, cnt;

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

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

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

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