UVALive - 3713 - Astronauts（图论——2-SAT）

Problem   UVALive - 3713 - Astronauts

Time Limit: 3000 mSec

Problem Description

Input

The input contains several blocks of test cases. Each case begins with a line containing two integers 1 ≤ n ≤ 100000 and 1 ≤ m ≤ 100000. The number n is the number of astronauts. The next n lines specify the age of the n astronauts; each line contains a single integer number between 0 and 200. The next m lines contains two integers each, separated by a space. A line containing i and j (1 ≤ i,j ≤ n) means that the i-th astronaut and the j-th astronaut hate each other. The input is terminated by a block with n = m = 0.

Output

For each test case, you have to output n lines, each containing a single letter. This letter is either ‘A’, ‘B’, or ‘C’. The i-th line describes which mission the i-th astronaut is assigned to. Astronauts that hate each other should not be assigned to the same mission, only young astronauts should be assigned to Mission B and only senior astronauts should be assigned to Mission A. If there is no such assignment, then output the single line ‘No solution.’ (without quotes).

Sample Input

16 20 21 22 23 24 25 26 27 28 101 102 103 104 105 106 107 108 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 1 10 2 9 3 12 4 11 5 14 6 13 7 16 8 15 1 12 1 13 3 16 6 15 0 0

Sample Output

B C C B C B C B A C C A C A C A

题解：几乎是2-SAT板子题，节点之间的关系稍微复杂了一点，总体来说并不困难。两种人，老年人只能做A，C，年轻人只能做B，C，所以每种人分别对应两种人格（其实就是true和false）如果相互厌恶的两人同类，那么不能同为true，也不能同为false，如果不同类，就只需不同为false（对应C）即可，至于输出方案，mark数组就是答案。

 #include <bits/stdc++.h>

 using namespace std;

 #define REP(i, n) for (int i = 1; i <= (n); i++)
 #define sqr(x) ((x) * (x))

 const int maxn =  + ;
 const int maxm =  + ;
 const int maxs =  + ;

 typedef long long LL;
 typedef pair<int, int> pii;
 typedef pair<double, double> pdd;

 const LL unit = 1LL;
 const int INF = 0x3f3f3f3f;
 const LL mod = ;
 const double eps = 1e-;
 const double inf = 1e15;
 const double pi = acos(-1.0);

 struct TwoSAT
 {
     int n, mark[maxn * ];
     vector<int> G[maxn * ];
     int S[maxn * ], c;

     void init(int n)
     {
         this->n = n;
         memset(mark, , sizeof(mark));
         for (int i = ; i <  * n; i++)
         {
             G[i].clear();
         }
     }

     bool dfs(int x)
     {
         if (mark[x ^ ])
             return false;
         if (mark[x])
             return true;

         mark[x] = true;
         S[c++] = x;
         for (auto v : G[x])
         {
             if (!dfs(v))
                 return false;
         }
         return true;
     }

     bool solve()
     {
         for (int i = ; i <  * n; i += )
         {
             if (!mark[i] && !mark[i + ])
             {
                 c = ;
                 if (!dfs(i))
                 {
                     while (c > )
                     {
                         mark[S[--c]] = ;
                     }
                     if (!dfs(i + ))
                         return false;
                 }
             }
         }
         return true;
     }

     void add_clause(int x, int xval, int y, int yval)
     {
         x = x *  + xval;
         y = y *  + yval;
         G[x ^ ].push_back(y);
         G[y ^ ].push_back(x);
     }
 };

 int n, m, sum;
 int age[maxn];
 TwoSAT solver;

 bool is_young(int x)
 {
     return x * n < sum;
 }

 main()
 {
     ios::sync_with_stdio(false);
     cin.tie();
     //freopen("input.txt", "r", stdin);
     //freopen("output.txt", "w", stdout);
     while (cin >> n >> m && (n || m))
     {
         solver.init(n);
         sum = ;
         for (int i = ; i < n; i++)
         {
             cin >> age[i];
             sum += age[i];
         }
         int u, v;
         for (int i = ; i < m; i++)
         {
             cin >> u >> v;
             if (u == v)
                 continue;
             u--, v--;
             if (is_young(age[u]) == is_young(age[v]))
             {
                 solver.add_clause(u, , v, );
                 solver.add_clause(u, , v, );
             }
             else
             {
                 solver.add_clause(u, , v, );
             }
         }
         if (solver.solve())
         {
             for (int i = ; i <  * n; i += )
             {
                 int a = age[i / ];
                 if (a * n >= sum)
                 {
                     if (solver.mark[i])
                     {
                         cout << "C" << endl;
                     }
                     else
                     {
                         cout << "A" << endl;
                     }
                 }
                 else
                 {
                     if (solver.mark[i])
                     {
                         cout << "C" << endl;
                     }
                     else
                     {
                         cout << "B" << endl;
                     }
                 }
             }
         }
         else
         {
             cout << "No solution." << endl;
         }
     }
     return ;
 }