題目大意

給出n,a,b和n個數(shù)，有兩個集合A,B，如果x放在A中則a-x必須存在且在A中，B同理，問你是否有合法方案

---

解題思路

不難發(fā)現(xiàn)，如果a-x或b-x存在，那么和x必須在同一個集合（若x放A中b-x放B中則不滿足題面的性質(zhì)）

那么把有關聯(lián)的點用并查集連一起，并判斷是否都存在a-x或b-x

---

code

#include<map> #include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define ll long long #define N 100100 using namespace std; ll T,n,a,b,pp,to[N][3],v[N][3],s[N],fa[N]; map<ll,ll>p; ll find(ll x) {return fa[x]==x?x:fa[x]=find(fa[x]); } void merge(ll x,ll y) {ll xx=find(x),yy=find(y);if(xx==yy)return;fa[xx]=yy;v[yy][1]&=v[xx][1];v[yy][2]&=v[xx][2];return; } int main() {scanf("%lld",&T);while(T--){scanf("%lld%lld%lld",&n,&a,&b);for(ll i=1;i<=n;++i){fa[i]=i;scanf("%lld",&s[i]);p[s[i]]=i;}for(ll i=1;i<=n;++i){if(s[i]>=a)to[i][1]=0;else to[i][1]=p[a-s[i]];//a-xif(s[i]>=b)to[i][2]=0;else to[i][2]=p[b-s[i]];v[i][1]=(to[i][1]?1:0);//連邊v[i][2]=(to[i][2]?1:0);}for(ll i=1;i<=n;++i){if(to[i][1])merge(i,to[i][1]);if(to[i][2])merge(i,to[i][2]);}pp=1;for(ll i=1;i<=n;++i)if(fa[i]==i)if(!(v[i][1]||v[i][2]))pp=0;//必須可以同時放到一個集合中if(pp)puts("YES");else puts("NO");for(ll i=1;i<=n;++i)p[s[i]]=0;}return 0; }

總結

以上是生活随笔為你收集整理的【并查集】公司搬迁的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。