校内测 10.29 T2 公司搬迁【并查集】

---

题目：

传送门

---

题意：

有$n$个员工，我们需要将他们分为两组，一个员工可以到一组，当且仅当$a-x$或$b-x$也在该组，对于一个$n$，我们需要知道是否有合法解

---

分析：

因为一个员工非零即一，所以可以开$2\ast n$的并查集，$i$和$i+n$分别表示在的那个组和不在的那个组
对于一个员工$x$，如果$a-x、b-x$都不存在，那么肯定是不合法的；如果两个都存在，我们不好判断是哪个和$x$放在了一起，但可以肯定，他们两个是一定在不同组的，所以把$a-x、b-x$连起来，$n+a-x、n+b-x$连起来，如果只有其一存在，那么$x$肯定是和存在的那个放在一起
最后方案合法性，我们就判断$x$和$n+x$是否在同一集合，如果在，那么明显不合法

---

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>  
#include<string>
#include<algorithm>
#include<queue>
#include<ctime>
#include<map>
#define LL long long
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
int f[200005],p[100005];
int find(int i) {return f[i]==i?i:f[i]=find(f[i]);}
map<int,int> m;
int main()
{
	int t=read();
	while(t--)
	{
		int n=read(),a=read(),b=read();
		for(int i=1;i<=n;i++) m[p[i]=read()]=i;
		for(int i=1;i<=2*n;i++) f[i]=i;
		int tf=0;
		for(int i=1;i<=n;i++)
		{
			if(!m[a-p[i]]&&!m[b-p[i]]) {tf=1;break;}
			int x=m[a-p[i]],y=m[b-p[i]];
			if(!x) 
			{
				f[i]=find(y);
				f[i+n]=find(y+n);
				continue;
			}
			if(!y) 
			{
				f[i]=find(x);
				f[i+n]=find(x+n);
				continue;
			}
			if(x==i||y==i) continue;
			f[x]=find(y+n);
			f[y]=find(x+n);
		} 
		if(tf) {printf("NO\n");continue;}
		for(int i=1;i<=n;i++) if(find(i)==find(n+i)) {tf=1;break;};
		if(tf) printf("NO\n"); else printf("YES\n");
	}
	return 0;
}