codeforces 468B two set(并查集)
Posted tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了codeforces 468B two set(并查集)相关的知识,希望对你有一定的参考价值。
链接
题意
给两个集合A B,两个数a b,n个数x,分配n个数到两个集合,要求x , a-x在同一个集合,x , b-x在同一个集合,属于A集合的数输出0,B的输出1,无解输出NO,空集也算一个集合,所以存在所有数都在一个集合。
做法
这有一个样例
input
6 5 9
1 2 3 4 5 8
output
YES
1 0 0 1 1 1
[1 4]A,[4 5]B,[1,8]B,所以只能把1 4都放进B集合
首先,能在A里面处理的就放入A,否则放入B,能在B里面处理的就放入B,否则放入A,如果x只能在B出现然而b-x在A里面则把b-x拿来B,你是不是担心把b-x从A里拿掉了,那a-(b-x)岂不是没有配对了?如果是这样的话那就是不合法的呗,不然后面还会把a-(b-x)从A中拿给B的,如上面的样例。如果两个集合里有重复元素则不合法。
代码
#include<bits/stdc++.h>
using namespace std;
#define M 111000
int fa[M], num[M];
int find(int x) {
return fa[x] == x ? x : fa[x] = find(fa[x]);
}
int main() {
int n, a, b;
while (cin >> n >> a >> b) {
map<int, int> mp;
for (int i = 1; i <= n; i++) {
cin >> num[i];
mp[num[i]] = i;
}
for (int i = 1; i <= n + 2; i++)
fa[i] = i;
int x;
for (int i = 1; i <= n; i++) {
x = find(i);
if (mp[a - num[i]]) fa[x] = find(mp[a - num[i]]);//如果a-num[i]存在的话,则把num[i]和a-num[i]放进a集合
else fa[x] = find(n + 1);//否则放在b集合
x = find(i);//为什么不直接用上面的x,因为合并后fa[x]已经变化了
if (mp[b - num[i]]) fa[x] = find(mp[b - num[i]]);//如果b-num[i]存在的话,则把num[i]和b-num[i]放进b集合
else fa[x] = find(n + 2);//否则放在a集合
}
if (find(n + 1) == find(n + 2))//如果两个集合有重复元素则不合法
cout << "NO\\n";
else {
cout << "YES\\n";
printf("%d", find(1) == find(n + 1));
for (int i = 2; i <= n; i++)
printf(" %d", find(i) == find(n + 1));
cout << endl;
}
}
return 0;
}
以上是关于codeforces 468B two set(并查集)的主要内容,如果未能解决你的问题,请参考以下文章