codeforces B. Coach 解题报告

题目链接：http://codeforces.com/problemset/problem/300/B

题目意思：给出n个students（n%3 = 0），编号依次为1～n，接下来有m行，每行有两个数：a和b（1<=a, b <= n），表示号码a的人想和号码b的人一起组队。但最终每一组的人最多只能为3个，至于组里只有2个或根本不成组的人可以通过组队变成3个人。问最后每组是否恰好为3人（包括原来不需要组队和组队后的情况），是则输出每组人的编号，否则输出-1。

一开始做的时候觉得像并查集，后来觉得做起来特别复杂，看了看tutorial，知道哪些情况为输出-1的：

情况1：组里面超过3人    情况2：2人组人数 > 1人组的人数（不成组）

根据这样硬用并查集来做，代码量非一般多！！！这就是不会算法的悲剧咯～～～

我的思路：

先用并查集来连边，保证大编号的指向最小编号的。

接着统计一人组，二人组，三人组的人数（分别为c1，c2，c3），以判断结果为-1的两种情况。

最后依次输出： 二人组 + 一人组，三人组，一人组 （除了三人组的情况其他两种都要组合）

（以后学好dfs一定要写条简短的代码，呜呜呜～～～整个晚上没了）

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

 const int maxn =  + ;
 int res[maxn], cnt[maxn], vis[maxn], vis1[maxn], vis2[maxn];

 int find(int x)
 {
     while (x != res[x])
         x = res[x];
     return x;
 }

 void merge(int x, int y)
 {
     int fx = find(x);
     int fy = find(y);
     if (fx > fy)
         res[fx] = fy;
     else
         res[fy] = fx;
 }

 int main()
 {
     int n, m, i, j, k, a, b;
     while (scanf("%d%d", &n, &m) != EOF)
     {
         for (i = ; i <= n; i++)
             res[i] = i;
         for (i = ; i < m; i++)
         {
             scanf("%d%d", &a, &b);
             merge(a, b);
         }
         memset(vis, , sizeof(vis));
         memset(vis1, , sizeof(vis1));
         memset(cnt, , sizeof(cnt));
         int flag, c2, c1, c3, f;
         c1 = c2 = c3 = ;
         for (i = ; i <= n; i++)
         {
             flag = ;
             if (res[i] == i)
             {
                 f = ;
                 for (j = i+; j <= n; j++)
                 {
                     if (res[j] == i)
                     {
                         cnt[i]++;
                         f = ;    // 1 p num 的标记
                     }
                 }
                 if (cnt[i] == )  // 3 p num
                     c3++;
                 if (cnt[i] == )  // 2 p num
                 {
                     c2++;
                     vis1[i] = ;
                 }
                 if (cnt[i] >= )    // 组里的人多于3个
                     flag = ;
                 if (!f)            //  1 p num
                 {
                     vis[i] = ;
                     c1++;
                 }
             }
             if (!flag)
                 break;
         }
         if (c2 > c1 || !flag)
             printf("-1\n");
         else
         {
             memset(vis2, , sizeof(vis2));
             for (i = ; i <= n; i++)     // 2 p num + 1 p num
             {
                 if (res[i] == i && vis1[i])  // vis1[i] 保证是2 p num
                 {
                     printf("%d ", i);
                     vis2[i] = ;        // sign used
                     for (j = i+; j <= n; j++)
                     {
                         if (!vis[j] && res[j] == i)  // !vis[j]：保证先输出2 p num的人
                         {
                             vis2[j] = ;
                             printf("%d ", j);
                             for (k = ; k <= n; k++)  // 1 p
                             {
                                 if (res[k] == k && vis[k] && !vis2[k])  // vis[k]:1 p num 的人
                                 {
                                     printf("%d\n", k);
                                     vis2[k] = ;
                                     c1--;  // 每输出一次减去一个1 p num的人
                                     break;
                                 }
                             }
                         }
                     }
                 }
             }
             for (i = ; i <= n; i++)     // 3 p num
             {
                 if (res[i] == i && !vis1[i] && !vis[i]) // !vis1[i]：保证不是2 p num；!vis[i]：保证不是1 p num
                 {
                     for (j = i; j <= n; j++)
                     {
                         if (res[j] == i && !vis[j])
                         {
                             printf("%d ", j);
                             vis2[j] = ;
                         }
                     }
                     printf("\n");
                 }
             }
             int count = ;
             for (i = ; i <= n && c1; i++)    // 剩下的 1 p num的人，每3个人合成1组
             {
                 if (!vis2[i])
                 {
                     printf("%d ", i);
                     c1--;
                     count++;
                     if (count %  == )
                         printf("\n");
                 }
             }
         }
     }
     return ;
 }