二分圖思想 把n個數分配到兩個集合

 

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<cmath>
#include<string>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
//二分圖思想 
//vis[i]=-1:num[i]還未分配到集合
//vis[i]=0:num[i]分配到集合A
//vis[i]=1:num[i]分配到集合B 
int vis[100001];
int num[100001];
//mp[num[i]=i;num[i]的位置是i 
map<int ,int >mp;
int A,B,n;
//判斷u能否被分配到集合m,如果能順便更新該位置的vis值 
bool dfs(int u,int m,int k){
	int p=mp[u];//記錄該數的位置 
	if(!mp[m-u])return false;//如果在此集合與此數匹配的數不存在。返回false 
	int q=mp[m-u];//如果存在，記錄位置 
	if(vis[q]==-1||u*2==m){//如果匹配的數未在任何集合中或者就是本身 
		vis[p]=vis[q]=k;//直接加入集合m
		return true;
	}//否則該匹配的數已經匹配過並且在另一個集合中（之前不可能在本集合中） 
	else if(dfs(A+B-2*m+u,m,k)){
		//A+B-2*m+u=(A-m)+B-(m-u) or (B-m)+A-(m-u)
		//以上數值爲匹配點m-u在另一個集合的匹配點的數值
		//也就是判斷能否將在另一個集合已經匹配過的以上兩點移動到當前集合
		//也就是判斷點（A+B-2*m+u）能否在當前集合中，如果能說明m-u可以移到當前集合，即匹配成功 
		vis[q]=vis[p]=k;
		return true;
	}
	else return false;//否則返回失敗 
}

bool solve(){
	memset(vis,-1,sizeof(vis));
	for(int i=1;i<=n;i++){
		if(vis[i]==-1){//每次尋找未在集合中的點 
			//如果即不能再集合A中也不能在集合B中，則返回false； 
			if(!dfs(num[i],A,0)&&!dfs(num[i],B,1)){
				return false;
			}
		}
	}
	return true;
}

int main(){
	cin>>n>>A>>B;
	for(int i=1;i<=n;i++){
		scanf("%d",num+i);
		mp[num[i]]=i;
	}
	if(solve()){
		//可以合法的分配 
		cout<<"YES"<<endl;
		for(int i=1;i<n;i++){
			printf("%d ",vis[i]);
		}
		printf("%d\n",vis[n]);
	}
	else cout<<"NO"<<endl;
	return 0;
}
//            /\       |  /  |**、
//			 /  \      | /   |   \
//			/    \     |/    |   /  _____                      ____   |  /
//		   /------\    |\    |__/  /     \  \      /\      /  /    \  | /
//		  /        \   | \   |    /       \  \    /  \    /  /______\ |/
//		 /          \  |  \  |    \       /   \  /    \  /   \        |
//      /            \ |   \ |     \_____/     \/      \/     \_____  |
/**
 *　　　　　　　　┏┓　　 　┏┓
 * 　　　　　　　┏┛┗━━━━━━━┛┗━━━┓
 * 　　　　　　　┃　　　　　　　┃ 　
 * 　　　　　　　┃　　　━　　 　┃
 * 　　　　　　　┃　＞　　　＜　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┃...　⌒　... 　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┗━┓　　　┏━┛
 * 　　　　　　　　　┃　　　┃　Code is far away from bug with the animal protecting　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃   神獸保佑,代碼無bug
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃  　　　　　　
 * 　　　　　　　　　┃　　　┃
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┗━━━┓
 * 　　　　　　　　　┃　　　　　　　┣┓
 * 　　　　　　　　　┃　　　　　　　┏┛
 * 　　　　　　　　　┗┓┓┏━┳┓┏┛
 * 　　　　　　　　　　┃┫┫　┃┫┫
 * 　　　　　　　　　　┗┻┛　┗┻┛
 */
// warm heart, wagging tail,and a smile just for you!
//
//                            _ooOoo_
//                           o8888888o
//                           88" . "88
//                           (| -_- |)
//                           O\  =  /O
//                        ____/`---'\____
//                      .'  \|     |//  `.
//                     /  \|||  :  |||//  \
//                    /  _||||| -:- |||||-  \
//                    |   | \\  -  /// |   |
//                    | \_|  ''\---/''  |   |
//                    \  .-\__  `-`  ___/-. /
//                  ___`. .'  /--.--\  `. . __
//               ."" '<  `.___\_<|>_/___.'  >'"".
//              | | :  `- \`.;`\ _ /`;.`/ - ` : | |
//              \  \ `-.   \_ __\ /__ _/   .-` /  /
//         ======`-.____`-.___\_____/___.-`____.-'======
//                            `=---='
//        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
//