NOI2015程序自动分析 并查集

在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本：假设 x1,x2,x3,… 代表程序中出现的变量，给定 n 个形如 xi=xj 或 xi≠xj 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

1≤n≤1000001≤i,j≤1000000000
题解:
考虑用并查集，若xi,xj相同，就用并查集连起来，判断哪些不同的，如果xi!=xj但是两者属于同一集合，就不行。由于i,j很大所以要用到离散化。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+10;
int cas,n1,n2,n,m;
struct node{
	ll x,y;
	int e;
}a[N][2];
ll b[N];
int fa[N];
int getfa(int x){
	if(x==fa[x]) return x;
	else return fa[x]=getfa(fa[x]);
}
int main(){
	scanf("%d",&cas);
	while(cas--){
		scanf("%d",&n);
		
		n1=n2=m=0;
		
		
		
		for(int i=1;i<=n;i++){
			ll u,v;int e;
			scanf("%lld%lld%d",&u,&v,&e);		
			b[++m]=u;b[++m]=v;
			if(e==1){
				a[++n1][0].x=u;a[n1][0].y=v;a[n1][0].e=1;
			}else{
				a[++n2][1].x=u;a[n2][1].y=v;a[n2][1].e=0;
			}
		}
		sort(b+1,b+m+1);
		int h=unique(b+1,b+m+1)-(b+1);
		for(int i=1;i<=h;i++) fa[i]=i; 
		for(int i=1;i<=n1;i++){
			int x1=lower_bound(b+1,b+h+1,a[i][0].x)-b;
			int y1=lower_bound(b+1,b+h+1,a[i][0].y)-b;
			int xx=getfa(x1);
			int yy=getfa(y1);
			if(xx!=yy){
				fa[xx]=yy;
			}	
		}
		bool flag=1;
		for(int i=1;i<=n2;i++){
			int x1=lower_bound(b+1,b+h+1,a[i][1].x)-b;
			int y1=lower_bound(b+1,b+h+1,a[i][1].y)-b;
			int xx=getfa(x1);
			int yy=getfa(y1);
			if(xx==yy) {
				flag=0;break;
			}
		}
		if(flag) printf("YES\n");
		else printf("NO\n");	
	} 
	return 0;
}