训练赛：约束判定

洛谷：程序自动分析原题
由题可知，并查集维护一下约数关系，$i，j$太大了离散化一下就可以了。裸题简单记录一下，（不懂的同学百度：并查集、离散化）。输入数据较多不要用cin，超时警告

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#define maxn 200005
#define INF 0x3f3f3f3f
#define mst(a) memset(a,0,sizeof a)
#define ll long long
using namespace std;
int f[maxn],lisa[maxn*2+10];
// 约束判定
struct node
{
	int x,y,e;	
}a[maxn];

bool cmp(node a,node b)
{
	return a.e>b.e;
}

int find(int x)
{
	if(x!=f[x])
		f[x] = find(f[x]);
	return f[x];
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		mst(lisa),mst(a),mst(f);
		int n,ok = 1;
		cin>>n;
		int id = -1;
		for(int i = 1; i<=n; i++)
		{
			scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].e);
			lisa[++id] = a[i].x;
			lisa[++id] = a[i].y;
		}
		sort(lisa,lisa+id);
		int res=unique(lisa,lisa+id)-lisa;  
        for(int i=1;i<=n;++i){
           a[i].x = lower_bound(lisa,lisa+res,a[i].x) - lisa;
           a[i].y = lower_bound(lisa,lisa+res,a[i].y) - lisa;   
        } 
		sort(a+1,a+n+1,cmp);
		for(int i = 1; i<=res; i++)f[i] = i;
		for(int i = 1; i <=n; i++)
		{
			int r1 = find(a[i].x);
			int r2 = find(a[i].y);
			if(a[i].e)
			{
				f[r1] = r2;
			}else if(r1 == r2)
			{
				printf("NO\n");
				ok = 0;
				break;
			}
		}
		if(ok)printf("YES\n");
	}

	return 0;
}