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+;
 typedef pair<int,int> pll;
 const int N = 1e5+;
 int vis[N], pi[N];
 map<int, int> m;
 int n, a, b;
 bool dfs(int v, int c){
     if(c == ){
         if(vis[m[b-v]] == -){
             vis[m[b-v]] = vis[m[v]] = ;
             return true;
         }
         if(vis[m[a-v]] == -){
             vis[m[a-v]] = vis[m[v]] = ;
             return true;
         }
         if(vis[m[a-v]] == ){
             vis[m[a-v]] = vis[m[v]] = ;
             return dfs(b-a+v,);
         }
         return false;
     }
     else if(c == ){
         if(vis[m[a-v]] == - || vis[m[a-v]] == ){
             vis[m[a-v]] = vis[m[v]] = ;
             return true;
         }
         if(vis[m[a-v]] == ){
             vis[m[a-v]] = vis[m[v]] = ;
             return dfs(b-a+v,);
         }
         return false;
     }
     return false;
 }
 int main(){
     ios::sync_with_stdio(false);
     cin.tie();
     cout.tie();
     cin >> n >> a >> b;
     vis[] = ;
     for(int i = ; i <= n; i++){
         vis[i] = -;
         cin >> pi[i];
         m[pi[i]] = i;
     }
     for(int i = ; i <= n; i++){
         if(vis[i] == -){
             if(dfs(pi[i],)) ;
             else {
             cout << "NO\n";
             return ;
             }
         }
     }
     cout << "YES\n";
     for(int i = ; i < n; i++)
         cout << vis[i] << ' ';
     cout << vis[n] << endl;
     return ;
 }