Codeforces 468 B Two Sets

Two Sets

题意：就是将一对数放进setA, setB中， 如果放进setA的话要求满足 x与a-x都在这个集合里面， 如果放进setB中要求满足x与b-x都在这个集合中。

题解：我们将能放进B的元素优先放在B中,如果能放进B就直接将2个元素放进B中。 然后如果能放进A中就放进A中， 如果在放入A过程中发现对应元素在setB中，那么就找对应元素的对应元素能不能放进A中。如果在元素放入B的过程中发现对应元素在A中，那么就无解了，因为优先放B中。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
typedef pair<int,int> pll;
const int N = 1e5+5;
int vis[N], pi[N];
map<int, int> m;
int n, a, b;
bool dfs(int v, int c){
    if(c == 2){
        if(vis[m[b-v]] == -1){
            vis[m[b-v]] = vis[m[v]] = 1;
            return true;
        }
        if(vis[m[a-v]] == -1){
            vis[m[a-v]] = vis[m[v]] = 0;
            return true;
        }
        if(vis[m[a-v]] == 1){
            vis[m[a-v]] = vis[m[v]] = 0;
            return dfs(b-a+v,1);
        }
        return false;
    }
    else if(c == 1){
        if(vis[m[a-v]] == -1 || vis[m[a-v]] == 0){
            vis[m[a-v]] = vis[m[v]] = 0;
            return true;
        }
        if(vis[m[a-v]] == 1){
            vis[m[a-v]] = vis[m[v]] = 0;
            return dfs(b-a+v,1);
        }
        return false;
    }
    return false;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> a >> b;
    vis[0] = 3;
    for(int i = 1; i <= n; i++){
        vis[i] = -1;
        cin >> pi[i];
        m[pi[i]] = i;
    }
    for(int i = 1; i <= n; i++){
        if(vis[i] == -1){
            if(dfs(pi[i],2)) ;
            else {
            cout << "NO
";
            return 0;
            }
        }
    }
    cout << "YES
";
    for(int i = 1; i < n; i++)
        cout << vis[i] << ' ';
    cout << vis[n] << endl;
    return 0;
}