73 lines
1.9 KiB
C++
73 lines
1.9 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
const int N = 200010; // 因为是需要存入左右两个条件xi=xj,这样最多是保存的两倍的n
|
||
|
||
int m; // m个条件
|
||
int b[N], bl; // 离散化数组，数组长度
|
||
int p[N]; // 并查集数组
|
||
struct Node {
|
||
int x, y, e;
|
||
} a[N]; // 输入的条件
|
||
|
||
// 并查集
|
||
int find(int x) {
|
||
if (x == p[x]) return x;
|
||
return p[x] = find(p[x]);
|
||
}
|
||
|
||
// 利用二分计算出x值在已排序数组b中的位置，位置就是新的号码
|
||
int get(int x) {
|
||
int l = 1, r = bl;
|
||
while (l < r) {
|
||
int mid = (l + r) >> 1;
|
||
if (b[mid] < x)
|
||
l = mid + 1;
|
||
else
|
||
r = mid;
|
||
}
|
||
return l;
|
||
}
|
||
|
||
int main() {
|
||
int T;
|
||
scanf("%d", &T);
|
||
while (T--) {
|
||
int flag = 0, idx = 0;
|
||
scanf("%d", &m);
|
||
|
||
for (int i = 1; i <= 2 * m; i++) p[i] = i;
|
||
|
||
for (int i = 1; i <= m; i++) {
|
||
scanf("%d %d %d", &a[i].x, &a[i].y, &a[i].e);
|
||
b[idx++] = a[i].x, b[idx++] = a[i].y; // 存入离散化数组中，准备处理
|
||
}
|
||
|
||
// 静态数组离散化
|
||
sort(b, b + 2 * m);
|
||
bl = unique(b, b + 2 * m) - b;
|
||
|
||
// 相等关系 <=> 同一个并查集
|
||
for (int i = 1; i <= m; i++)
|
||
if (a[i].e == 1) {
|
||
int pa = find(get(a[i].x)), pb = find(get(a[i].y));
|
||
if (pa != pb) p[pa] = pb;
|
||
}
|
||
|
||
// 不等关系 与 同一个并查集 存在冲突
|
||
for (int i = 1; i <= m; i++)
|
||
if (a[i].e == 0) {
|
||
int pa = find(get(a[i].x)), pb = find(get(a[i].y));
|
||
if (pa == pb) {
|
||
flag = 1;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (flag)
|
||
puts("NO");
|
||
else
|
||
puts("YES");
|
||
}
|
||
return 0;
|
||
}
|