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 = 100010;

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, -1, sizeof(ans));
    for(int i = 1; i < N; i++) {
        fa[i] = i;
    }
    scanf("%d%d%d", &n, &A, &B);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        mp[a[i]] = i;
    }
    for(int i = 1; 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 0;
        }
        else if(x) {
            int &t = ans[mp[B - a[i]]];
            if(t == 0) {
                printf("NO");
                return 0;
            }
            if(t == -1) {
                t = 1;
            }
            ans[i] = 1;
            merge(i, mp[B - a[i]]);
        }
        else if(y) {
            int &t = ans[mp[A - a[i]]];
            if(t == 1) {
                printf("NO");
                return 0;
            }
            if(t == -1) {
                t = 0;
            }
            ans[i] = 0;
            merge(i, mp[A - a[i]]);
        }
        else {
           merge(i, mp[A - a[i]]);
           merge(i, mp[B - a[i]]);
        }
    }

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

    return 0;
}