记录编号 561282 评测结果 AAAAAAAAAA
题目名称 [NOI 2015]程序自动分析 最终得分 100
用户昵称 Gravataryrtiop 是否通过 通过
代码语言 C++ 运行时间 2.472 s
提交时间 2021-07-01 13:45:59 内存使用 3.47 MiB
显示代码纯文本
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 2e5 + 5;
int pre[maxn],c[maxn];
struct que {
    int i,j,e;
    que() {
        i = j = e = 0;
    }
    bool operator < (const que& x) const{
        return e > x.e;
    }
}a[maxn];
int find(int x) {
    return x == pre[x] ? x : pre[x] = find(pre[x]);
}
void union_set(int r1,int r2) {
    pre[r2] = r1;
    return ;
} 
int n;
void work() {
    scanf("%d",&n);
    int cnt = 0;
    for(int i = 1;i <= n;++ i) {
        scanf("%d%d%d",&a[i].i,&a[i].j,&a[i].e);
        c[++ cnt] = a[i].i;
        c[++ cnt] = a[i].j;
    }
    sort(c + 1 , c + 1 + cnt);
    cnt = unique(c + 1 , c + 1 + cnt) - c - 1;
    for(int i = 1;i <= cnt;++ i) {
        pre[i] = i;
    }
    sort(a + 1 , a + 1 + n);
    for(int i = 1;i <= n;++ i) {
        int x = lower_bound(c + 1 , c + 1 + cnt , a[i].i) - c;
        int y = lower_bound(c + 1 , c + 1 + cnt , a[i].j) - c;
        int r1 = find(x),r2 = find(y);
        if(a[i].e) {
            if(r1 != r2)union_set(r1 , r2);
        }
        else {
            if(r1 == r2) {
                puts("NO");
                return ;
            }
        }
    }
    puts("YES");
    return ;
}
int main() {
    freopen("prog.in","r",stdin);
    freopen("prog.out","w",stdout);
    int t;
    scanf("%d",&t);
    while(t --) {
        work();
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}