CF 468B Two Sets

题意：

给定n个正整数与a，b两个集合，求一种方案使得这n个数恰好被分在这两个集合中且集合中无多余的数且若x在a中则A-x在a中，若x在b中则B-x在b中。

题意理解了我好半天...

解法1：并查集。

把x, A - x，B - x(如果不存在B - x，x就不能放B集合)放入同一个并查集即可。

实现时注意一些乱七八糟的东西，大力讨论即可。

 #include <cstdio>
 #include <map>
 #include <cstring>
 const int N = ;

 int a[N], ans[N], A, B, n, fa[N];
 std::map<int, int> mp;

 inline int find(int x) {
     if(x == fa[x]) {
         return x;
     }
     return fa[x] = find(fa[x]);
 }
 inline bool check(int x, int y) {
     return find(x) == find(y);
 }
 inline void merge(int x, int y) {
     fa[find(x)] = find(y);
     return;
 }

 int main() {
     memset(ans, -, sizeof(ans));
     for(int i = ; i < N; i++) {
         fa[i] = i;
     }
     scanf("%d%d%d", &n, &A, &B);
     for(int i = ; i <= n; i++) {
         scanf("%d", &a[i]);
         mp[a[i]] = i;
     }
     for(int i = ; i <= n; i++) {
         bool x = mp.find(A - a[i]) == mp.end();
         bool y = mp.find(B - a[i]) == mp.end();
         if(x && y) {
             printf("NO");
             return ;
         }
         else if(x) {
             int &t = ans[mp[B - a[i]]];
             if(t == ) {
                 printf("NO");
                 return ;
             }
             if(t == -) {
                 t = ;
             }
             ans[i] = ;
             merge(i, mp[B - a[i]]);
         }
         else if(y) {
             int &t = ans[mp[A - a[i]]];
             if(t == ) {
                 printf("NO");
                 return ;
             }
             if(t == -) {
                 t = ;
             }
             ans[i] = ;
             merge(i, mp[A - a[i]]);
         }
         else {
            merge(i, mp[A - a[i]]);
            merge(i, mp[B - a[i]]);
         }
     }

     for(int i = ; i <= n; i++) {
         if(ans[i] == -) {
             ans[i] = ans[find(i)];
         }
         else {
             if(ans[find(i)] == -) {
                 ans[find(i)] = ans[i];
             }
             else if(ans[find(i)] != ans[i]) {
                 printf("NO");
                 return ;
             }
         }
     }
     printf("YES\n");
     for(int i = ; i <= n; i++) {
         if(ans[find(i)] == -) {
             printf("0 ");
         }
         else {
             printf("%d ", ans[find(i)]);
         }
     }

     return ;
 }

AC代码