CODEFORCES 468B Two Sets ---并查集建树

题意：把n个不同的数分成2个集合。

分析：分成两棵树。若两棵树连在一起，这说明有个点共用或都不满足a，b的关系，则输出NO；

样例：

Input

4 5 9
2 3 4 5

Output

YES
0 0 1 1

Input

3 3 4
1 2 4

Output

NO

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <map>

using namespace std;
const int MAXN = 1e5+10;
int pre[MAXN],arr[MAXN];
map<int,int> mp;
void init(int n)
{
    for(int i=1;i<n+3;i++)
        pre[i]=i;
}
int finds(int x)
{
    if(x==pre[x])
        return x;
    return pre[x] = finds(pre[x]);
}
void unions(int x,int y)
{
    int fx = finds(x);
    int fy = finds(y);
    if(fx!=fy)
    {
        pre[fx] = fy;
    }
    //printf("x= %d y =%d fx = %d fy = %d \n",x,y,fx,fy);
}
int main()
{
    int n,a,b;
    scanf("%d %d %d",&n,&a,&b);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&arr[i]);
        mp[arr[i]]=i;
    }
    init(n);
    for(int i=1;i<=n;i++)
    {
        if(mp[a-arr[i]])
            unions(i,mp[a-arr[i]]);
        else
            unions(i,n+2);
        if(mp[b-arr[i]])
            unions(i,mp[b-arr[i]]);
        else
            unions(i,n+1);
    }
    if(finds(n+1)==finds(n+2))
        printf("NO\n");
    else
    {
        printf("YES\n");
        for(int i=1;i<=n;i++)
        {
            if(i>1) printf(" ");
            if(finds(n+1)==finds(mp[arr[i]]))
                printf("0");
            else
                printf("1");
        }
        printf("\n");
    }
    return 0;

}