Codeforces Round #268 (Div. 2) D:Two Sets 并查集

　D:Two Sets

题意：将n个数分成两个集合。 如果x属于集合A，那么a - x一定属于集合A，如果x属于集合B，那么b - x一定属于集合B

首先很容易想到将，x与a - x同时存在，那么将它们放在A集合，同理B

但是这是错的，举个例子，a=3,b=6,   x: 1 2 4， x = 1 , a - x = 2 ， 所以 1 、2  放入集合A ，x = 4 ， b - 4 = 2  ， 那么2 ， 4在B集合 ，

如果a - x不存在，那么x一定不在A集合。

为什么？？ 因为，在A中没有与之相对应的a - x。

同样。。如果b -  x不存在，那么x一定不在B集合里面。

如果a - x不存在，可以将 x 放在B集合，如果b -  x不存在将 x 放在A集合，

如果一个x既不属于A也不属于B，那么这个x将会和集合A合并也会和集合B合并，即 A==B

如果一个x既属于A也属于B，那么a - x将会和集合B合并 ，b - x 和集合B合并, 而a - x，b - x ， x 是同一个集合，所以集合A==B 

所以A==B输出NO，

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 +100;
int n, a, b;
int f[N];
int p[N];
map<int, int> mp;
 
int find(int x){
    if(f[x] == -1) return x;
    else return f[x] = find(f[x]);
}
 
void unity(int x,int y){
    x = find(x), y = find(y);
    if(x == y) return ;
    f[x] = y;
}
int main(){
    cin >> n >> a >> b;
    memset(f, -1, sizeof(f));
    for(int i = 1; i <= n; i++){
        cin >> p[i];
        mp[p[i]] = i;
    }
     
    for(int i = 1; i <= n; i++){
        if(mp[a - p[i]]){
            unity(i, mp[a-p[i]]);
        }
        else unity(i, n + 1);//B
        if(mp[b - p[i]]){
            unity(i, mp[b - p[i]]);
        }
        else unity(i, n + 2);//A
    }
     
    if(find(n + 1) == find(n + 2)){
        cout << "NO\n";
    }
    else{
        cout << "YES\n";
        for(int i = 1; i <= n; i++){
            if(find(i) == find(n + 2)){
                cout<<(i == n ? "0\n" : "0 ");
            }
            else{
                cout<<(i == n ? "1\n" : "1 ");
            }
        }
    }
    return 0;
}