程序自动分析 并查集+离散化

题目：

代码：

#include<bits/stdc++.h>
using namespace std;

const int N=200010;

int p[N];
int m,n;

struct Query{
    int a,b,e;
}query[N];

unordered_map<int,int>h;

int get(int x){
    if(h.count(x)==0)return h[x]=n++;
    return h[x];
}

int find(int x){
    if(x!=p[x])p[x]=find(p[x]);
    return p[x];
}

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        h.clear();
        n=0;
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            int a,b,e;
            scanf("%d%d%d",&a,&b,&e);
            query[i]={get(a),get(b),e};
        }
        for(int i=0;i<2*m;i++)p[i]=i;
        for(int i=0;i<m;i++){
            int a=query[i].a,b=query[i].b,e=query[i].e;
            int pa=find(a),pb=find(b);
            if(e==1) p[pa]=pb;
        }
        bool has_conflict=false;
        for(int i=0;i<m;i++){
            int a=query[i].a,b=query[i].b,e=query[i].e;
            int pa=find(a),pb=find(b);
            if(e==0&&pa==pb){
                has_conflict=true;
                break;
            }
        }
        if(has_conflict)printf("NO\n");
        else printf("YES\n");
        
    }

    return 0;
}