HDU 5285 wyh2000 and pupil (二分图)

题意：共有n个小学生，编号为1−n。将所有小学生分成2组，每组都至少有1个人。但是有些小学生之间并不认识，而且如果a不认识b，那么b也不认识a。Wyh2000希望每组中的小学生都互相认识。而且第一组的人要尽可能多。请你帮wyh2000求出第一组和第二组的人数是多少。如果找不到分组方案，则输出"Poor wyh"。

思路：二分图着色。给的就是无向图，每次都累加人多的颜色即可。若不能着色，必定不能分成2组。如果全部都是1个颜色，那么要让其中1人过第2组。我勒个去，就因为赭色时颜色号码开小了而错。

  1 //#include <bits/stdc++.h>
  2 #include <cstdio>
  3 #include <cstring>
  4 #include <iostream>
  5 #include <map>
  6 #include <set>
  7 #include <deque>
  8 #include <algorithm>
  9 #include <vector>
 10 #define LL long long
 11 #define pii pair<int,int>
 12 #define INF 0x7f7f7f7f
 13 using namespace std;
 14 const int N=100000+100;
 15
 16 vector<int> vect[N];
 17 int col[N];
 18 set<int> mapp;
 19
 20 int color(int u, int f)
 21 {
 22     mapp.clear();
 23     deque<int> que;
 24     que.push_back(u);
 25     col[u]=f;
 26     mapp.insert(u);
 27     while(!que.empty())
 28     {
 29         int x=que.front();
 30         que.pop_front();
 31
 32         for(int i=0; i<vect[x].size(); i++)
 33         {
 34             int tmp=vect[x][i];
 35             if(col[tmp]==col[x]) return 0;  //冲突
 36             if(col[tmp]==0)   //没色
 37             {
 38                 mapp.insert(tmp);//一个连同分量装进去
 39                 col[tmp]=1000000-col[x];
 40                 que.push_back(tmp);
 41             }
 42         }
 43     }
 44     return 1;
 45 }
 46
 47
 48 int cal(int n)
 49 {
 50     memset(col,0,sizeof(col));
 51     int big=0, small=0, k=0;
 52     for(int i=1; i<=n; i++)
 53     {
 54         if(!col[i])
 55         {
 56             if(!color(i, ++k)) return 0;
 57
 58             //统计人数
 59             int a=0, b=0;
 60             set<int>::iterator it=mapp.begin();
 61             for(int j=0; j<mapp.size(); j++)
 62             {
 63                 if(col[*it]==k)   a++;
 64                 if(col[*it]==1000000-k) b++;
 65                 it++;
 66             }
 67
 68             if(a<b) swap(a, b);
 69             big+=a;
 70             small+=b;
 71         }
 72     }
 73     if(!small)  return big-1;
 74     else return big;
 75 }
 76
 77
 78
 79 int main()
 80 {
 81     freopen("input.txt", "r", stdin);
 82     int t, n, m, a, b;
 83     cin>>t;
 84     while(t--)
 85     {
 86         scanf("%d%d",&n,&m);
 87         for(int i=0; i<=n; i++)    vect[i].clear();
 88
 89         for(int i=0; i<m; i++)
 90         {
 91             scanf("%d%d",&a,&b);
 92             vect[a].push_back(b);
 93             vect[b].push_back(a);
 94         }
 95         if(n<=1)
 96         {
 97              puts("Poor wyh");
 98              continue;
 99         }
100         else if(n==2)
101         {
102              puts("1 1");
103              continue;
104         }
105
106         int tmp=cal(n);
107         if(tmp) printf("%d %d\n",tmp, n-tmp);
108         else  puts("Poor wyh");
109
110
111     }
112     return 0;
113 }

AC代码