CF 469B

構造出兩個節點n+1,n+2來存放A集合和B集合中的數據，顯然一個合理的分配不會使得一個元素既在A里面，也在B里面。而由于每一個元素都要去分配，如果a-x沒有那它就得在B里面（和n+2合并），同理對于b-x沒有的情況。最后根據它和n+1的祖宗是不是同一個來判斷是屬于哪個集合。

//CodeForces 469D #include <cstdio> #include <cstring> #include <string> #include <map> #include <iostream> #include <cmath> #include <vector> #include <set> #include <algorithm> typedef long long ll; #define rep(i,a,b) for (int i = (a); i <= (b); ++i) #define rep(i,a,b) for (int i = (a); i <= (b); ++i) using namespace std; const int N = 200010; int ans, n ,x, y, a[N], b[N], f[N]; map<int,int> mp; int find(int k){if (f[k]==k)return k;elsereturn find(f[k]); } void merge(int x,int y){int fx, fy;fx = find(x);fy = find(y);if (fx < fy)f[fy] = fx;elsef[fx] = fy; } int main() {cin >> n >> x >> y;rep (i, 1, n){scanf("%d",&a[i]);mp[a[i]] = i;}rep (i,1,n+2){f[i] = i;}rep (i, 1, n){if (mp.count(x-a[i])==1){//如果存在都在A集合，那么這兩個節點合并merge(i,mp[x-a[i]]);}else{//否則合并到n+1(B)集合里merge(i,n+1);}if (mp.count(y-a[i])==1){//都在B集合merge(i,mp[y-a[i]]);}else{merge(i,n+2);//合并到n+2（A）集合里}}if (find(n+1) == find(n+2)){//如果成立則這兩個點應該不會沖突puts("NO");return 0;}puts("YES");int anc = find(n+1);rep (i,1,n){//cout << find(i) << endl;cout << (find(i)==anc) << " ";}puts("");return 0; }

?

轉載于:https://www.cnblogs.com/ohazyi/p/7018454.html

總結

以上是生活随笔為你收集整理的并查集做题总结的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。