AcWing 237. 程序自动分析

输入样例：

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出样例：

NO
YES

 解析：

        并查集，由于 i, j 数据量为1e9，所以需要离散化。

        数组要开2e5个，由于可能每条“约束条件”的数据都不一样。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int t,n,s[N],idx,flag;
map<int,int>mp;
int func(int x){
	if(mp.count(x)) return mp[x];
	return mp[x]=++idx;
}
int find(int x){
	if(x!=s[x]) s[x]=find(s[x]);
	return s[x];
}
void merge(int x,int y){
	x=find(x),y=find(y);
	if(x!=y) s[x]=y;
}
struct node{
	int x,y,e;
}a[N];
int main(){
	scanf("%d",&t);
	while(t--){
		flag=1,idx=0;
		mp.clear();
		scanf("%d",&n);
		for(int i=1;i<=200000;i++) s[i]=i;
		for(int i=0;i<n;i++){
			int x,y,e;
			scanf("%d%d%d",&x,&y,&e);
			x=func(x);
			y=func(y);
			if(e) merge(x,y);
			a[i]={x,y,e};
		}
		for(int i=0;i<n;i++){
			if(!a[i].e)
			if(find(a[i].x)==find(a[i].y)){
				flag=0;
				cout<<"NO"<<endl;
				break;
			}
		}
		if(flag) cout<<"YES"<<endl;
	}
	return 0;
}