【并查集】程序自动分析

原题链接

规模太大，先用离散化缩小。

①考虑所有相等条件，把所有相等的变量捆绑进同一个集合中。
②考虑所有不等条件，若两个变量处于同一集合中，则证明无法同时满足这两个条件。

代码实现：

#include<bits/stdc++.h>
#define N 100000+10
using namespace std;

int n,m;
int a[2*N],fa[2*N];

int xi[N],xj[N];
bool e[N];

int get(int x){
	if(fa[x]==x)return x;
	return fa[x]=get(fa[x]);
}

int find(int x){
	return lower_bound(a+1,a+m+1,x)-a;
}

void merge(int x,int y){ 
	fa[get(x)]=get(y);
} 

void work(){
	cin>>n;
	
	for(int i=1;i<=n;i++){
		cin>>xi[i]>>xj[i]>>e[i];
		a[2*i-1]=xi[i];a[2*i]=xj[i];
	}
	
	sort(a+1,a+2*n+1);
	m=unique(a+1,a+2*n+1)-(a+1);
	
	for(int i=1;i<=m;i++)fa[i]=i;
	for(int i=1;i<=n;i++)
		if(e[i])merge(find(xi[i]),find(xj[i]));
	
	for(int i=1;i<=n;i++)
		if(!e[i]&&get(find(xi[i]))==get(find(xj[i]))){
			puts("NO");
			return;
		}
	puts("YES");
}

int main(){
	int t;cin>>t;
	
	for(int i=1;i<=t;i++)work();
	return 0;
}