【Acwing】并查集 237. 程序自动分析

237. 程序自动分析 - AcWing题库

题意：

思路：

经典并查集维护关系传递性的题目，将相等的元素分为一类，然后询问的时候看这两个元素在不在同一连通块就好了

小trick：由于数据较大，因此需要离散化一下

Code：

#include <bits/stdc++.h>
using namespace std;
const int mxn=2e5+10;
#define int long long
unordered_map<int,int> mp;
struct ty{
    int a,b,w;
}q[mxn];
int n,x,y,w,tot=0;
int f[mxn];
int s(int x){
    if(mp[x]) return mp[x];
    return mp[x]=++tot;
}
void init(){
    tot=0;
    mp.clear();
}
int find(int x){
    return f[x]=(x==f[x])?x:find(f[x]);
}
void join(int u,int v){
    int f1=find(u),f2=find(v);
    if(f1!=f2) f[f1]=f2;
}
void solve(){
    init();
    cin>>n;
    for(int i=1;i<=2e5;i++) f[i]=i;
    for(int i=1;i<=n;i++){
        cin>>x>>y>>w;
        x=s(x),y=s(y);
        q[i].a=x,q[i].b=y,q[i].w=w;
        if(w==1) join(x,y);
    }
    int ok=1;
    for(int i=1;i<=n;i++){
        if(!q[i].w){
            if(find(q[i].a)==find(q[i].b)){
                ok=0;
                break;
            }
        }
    }
    if(ok) cout<<"YES"<<'\n';
    else cout<<"NO"<<'\n';
}
signed main(){
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int T=1;
    cin>>T;
    while(T--)solve();
    return 0;
}