codeforces 468B Two Sets 2-sat

题意：给定n,a,b，n个不同的数，希望把它们分到A、B两个集合。如果x属于A，那么必须 a-x 也属于A。 如果x属于B，那么b-x必须

属于B。问时候能将n个数分到A、B两个集合中，如果能输出n个数属于哪个集合，为0属于A，为1属于B。

思路：我们很容易联想到2-sat。我们首先把每个数map映射一下，对于第i个数，用2＊i表示 i在A集合为真，2*i+1表示i在A集合为

假。令当前考虑第i个数，设u=i。如果mp[a-x[i]]不存在，那么推得如果此时i在A集合，那么i一定在B集合，即addedge(2*u,2*u+1)；

如果v=mp[ a - x[ i ] ]存在，那么推得如果u在A集合，那么v一定在A集合，即addedge(2*u,2*v)，同时推得如果v在B集合，那么u一定

在B集合(不然u不可能存在于A集合)，即addedge(2*v+1,2*u+1); 同理我们可以推得 mp[ b -x[ i ] ]的情况，不再赘述。详见代码：

// file name: codeforces468B.cpp //
// author: kereo //
// create time:  2014年09月21日 星期日 23时02分05秒 //
//***********************************//
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN=100000+100;
const int inf=0x3fffffff;
const int mod=1000000000+7;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
int n,a,b,edge_cnt,top;
int x[MAXN],mark[MAXN<<1],head[MAXN<<1],s[MAXN<<1];
map<int,int>mp;
struct Edge
{
	int v;
	int next;
}edge[MAXN<<2];
void init(){
	edge_cnt=0;
	memset(head,-1,sizeof(head));
	memset(mark,0,sizeof(mark));
}
void addedge(int u,int v){
	edge[edge_cnt].v=v;
	edge[edge_cnt].next=head[u];
	head[u]=edge_cnt++;
}
bool dfs(int x)  
{  
    if(mark[x^1]) return false;  
    if(mark[x]) return true;  
    mark[x]=true;  
    s[top++]=x;  
    for(int i=head[x];i!=-1;i=edge[i].next)  
    {  
        int v=edge[i].v;  
        if(!dfs(v))  
            return false;  
    }  
    return true;  
}  
bool solve()  
{  
    for(int i=0;i<2*n;i+=2)  
        if(!mark[i]&&!mark[i+1])  
        {  
            top=0;  
            if(!dfs(i))  
            {  
                while(top>0) mark[s[--top]]=false;  
                if(!dfs(i+1)) return false;  
            }  
        }  
    return true;  
}  
int main()
{
	while(~scanf("%d%d%d",&n,&a,&b)){
		mp.clear();
		for(int i=1;i<=n;i++){
			scanf("%d",&x[i]);
			mp[x[i]]=i;
		}
		int flag;
		init();
		for(int i=1;i<=n;i++){
			int val=x[i],u=i,v;
			u--;
			v=mp[a-val]; v--;
			if(!mp[a-val])
				addedge(2*u,2*u+1);
			else {
				addedge(2*u,2*v); 
				addedge(2*v+1,2*u+1);
			}
			v=mp[b-val]; v--;
			if(!mp[b-val])
				addedge(2*u+1,2*u);
			else{
				addedge(2*u+1,2*v+1); 
				addedge(2*v,2*u);
			}
		}
		flag=solve();
		if(!flag){
			printf("NO\n");
			continue;
		}
		printf("YES\n");
		for(int i=0;i<2*n;i+=2){
			if(i!=0)
				printf(" ");
			if(mark[i])
				printf("0");
			else 
				printf("1");
		}
		printf("\n");
	}
	return 0;
}