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代码