NOI2015 程序自动分析 prog

何等水题

某神犇仿关押罪犯的写法 却写挂了  然而实际上并不需要补集之类的

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std; 

const int Maxn=100010;
int n;
struct Data{
    int x,y;
    Data(int x=0,int y=0):x(x),y(y) {}
}q[Maxn][2];
int tot[2];
int seq[Maxn*2];
int hash(int x) {
    return lower_bound(seq+1,seq+n*2+1,x) - seq;
}
int fa[Maxn*2];
int find(int x) {
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
bool merge(int x,int y) {
    x=find(x);
    y=find(y);
    if(x==y) return 0;
    return fa[x]=y,1;
}

void init() {
    tot[0]=tot[1]=0;
    scanf("%d",&n);
    for(int i=1;i<=n*2;i++) fa[i]=i;
    int x,y,e;
    for(int i=1;i<=n;i++) {
        scanf("%d%d%d",&x,&y,&e);
        seq[i*2-1]=x;
        seq[i*2]=y;
        q[++tot[e]][e] = Data(x,y);
    }
    sort(seq+1,seq+n*2+1);
}

bool check() {
    for(int i=1;i<=tot[1];i++) {
        merge(hash(q[i][1].x),hash(q[i][1].y));
    }
    for(int i=1;i<=tot[0];i++) {
        int x=hash(q[i][0].x),y=hash(q[i][0].y);
        if(find(x)==find(y)) return 0;
    }
    return 1;
}

int main() {
    freopen("prog.in","r",stdin);
    freopen("prog.out","w",stdout);

    int T;
    for(scanf("%d",&T);T--;) {
        init();
        if(check())puts("YES");
        else puts("NO");
    }
    
    return 0;
}