P1955 map做法60分求调（闻灌佬多）

#include<bits/stdc++.h>
using namespace std;
int n,m,cnt=0;
unordered_map<long long,long long>fa;
struct pan{
    long long a;
    long long b;
}p[2000005];
long long find(long long x){
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
void he(long long x,long long y){
    long long xx=find(x),yy=find(y);
    if(xx==yy) return;
    fa[yy]=xx;
}
int main(){
    int T;
    cin>>T;
    while(T--){
        cin>>n;
        bool f=1;
        for(int i=1;i<=n;i++){
            long long x,y,e;
            scanf("%lld%lld%lld",&x,&y,&e);
            if(fa[x]==0){
                fa[x]=x;
            }
            if(fa[y]==0){
                fa[y]==y;
            }
            if(e==1){
                he(x,y);
            }
            else{
                p[++cnt].a=x;
                p[cnt].b=y;
            }
        }
        for(int i=1;i<=cnt;i++){
            long long xx=find(p[i].a),yy=find(p[i].b);
            if(xx==yy){
                f=0;
                break;
            }
        }
        if(f){
            printf("YES\n");
        }
        else{
            printf("NO\n");
        }
        cnt=0;
    }
    return 0;
}