codeforces 468B two set(并查集)

codeforces 468B two set(并查集)
n+1 表示 B 组
n+2 表示 A 组

按照 这题的做法
应该是 如果 满足 num[ i ] a-num[ i ] 则他们同一组 但不一定 就一定是 都是 A 组
也可能都是 B 组 然而如果不满足这个条件的话，就直接加入 B组
然后如果满足 num[ i ] b-num[ i ] 则加入同一组 然后不满足就 加入 A 组

#include <bits/stdc++.h> 
using namespace std ; 

const int maxn = 1e5+11 ; 
int n,a,b ; 
int num[maxn],fa[maxn] ;  

inline int find(int x) 
{
    if(fa[x] == x) return x ; 
    return fa[x] = find(fa[x]) ; 
}

int main() 
{
    scanf("%d%d%d",&n,&a,&b) ; 
    map<int,int> mp ; 
    for(int i=1;i<=n;i++) 
        scanf("%d",&num[ i ]) , mp[ num[ i ] ] = i ; 
    
    int x ; 
    for(int i=1;i<=n+2;i++) fa[ i ] = i ; 
    for(int i=1;i<=n;i++) 
    {
        x = find( i ) ; 
        if( mp.count( a-num[ i ] ) ) 
            fa[ x ] = find( mp[ a-num[ i ] ] ) ;  
        else  
            fa[ x ] = find( n+1 ) ;    //  不行的话就加入B 组 
        x = find( i ) ; 
        if( mp.count( b-num[ i ] ) ) 
            fa[ x ] = find( mp[ b-num[ i ] ] ) ; 
        else
            fa[ x ] = find( n+2 ) ;  
    }
    if( find(n+1)==find(n+2) ) 
    {
        printf("NO
") ;  
        return 0 ; 
    }
    printf("YES
") ; 
    for(int i=1;i<=n;i++) 
        printf("%d ",find( i )==find(n+1) ) ; 
    
    return 0 ; 
}