【SSL 1676】程序自动分析（并查集）

题目大意

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

考虑一个约束满足问题的简化版本：假设$x_1,x_2,x_3,\dots$代表程序中出现的变量，给定 $n$ 个形如$x_i=x_j$ 或 $x_i\ne x_j$ 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。

例如，一个问题中的约束条件为：$x_1=x_2，x_2=x_3，x_3=x_4，x_1\ne x_4$，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。
$1\le n\le 1000000$
$1\le i,j\le 1000000000$

输入格式
第$1$行包含$1$个正整数$t$，表示需要判定的问题个数，注意这些问题之间是相互独立的。

对于每个问题，包含若干行：

第$1$行包含$1$个正整数$n$，表示该问题中需要被满足的约束条件个数。

接下来$n$行，每行包括$3$个整数$i,j,e$，描述$1$个相等/不等的约束条件，相邻整数之间用单个空格隔开。若$e=1$，则该约束条件为$x_i=x_j$；若$e=0$，则该约束条件为$x_i\ne x_j$。

输出格式
输出包括$t$行。
第 $k$ 行输出一个字符串$“YES”$或者$“NO”$（不包含引号，字母全部大写），$“YES”$表示输入中的第$k$个问题判定为可以被满足，$“NO”$表示不可被满足。

输入样例

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

输出样例

NO
YES

基本思路

十分明显的并查集，我们把相同的存在一个并查集里，随后看是否有两个元素不相等但在同一集合内，这样就矛盾了。

这里的$i,j$这么大明显不可能开这么大的数组，而且其中有重复的数据，所以我们可以通过将其中所有数字都存起来，然后用$unique$ 进行排序去重，这样我们就可以得到没有重复的数据。然后操作的话就使用二分查找进行查询再合并，$n$ 总归是在数组可以承受的范围嘛，可以进行并查集操作。所以我们要先用一个结构体数组把每个操作记录下来，后面离散化完了再操作。

核心代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+10;
struct node{
	int x,y,p;
}a[N];
int t,n,m,fa[N],cnt[N],d,ans;
inline int find(int p){
	if(fa[p]==p) return p;
	return fa[p]=find(fa[p]);
}
inline void merge(int x,int y){
	x=find(x);y=find(y);
	if(x!=y) fa[x]=y;
}
inline void solve(){
	cin>>m;
	d=0;
	for(int i=1;i<=m;i++){
		cin>>a[i].x>>a[i].y>>a[i].p;
		cnt[++d]=a[i].x;
		cnt[++d]=a[i].y;
	}
	sort(cnt+1,cnt+1+d);
	//离散化 
	n=unique(cnt+1,cnt+1+d)-cnt-1;//从1开始计还要-1 
	//去重,但不是把重复的数去掉
	//而是放在数组后面
	//1 1 2 2 -> 1 2 2 2 所以注意下面的二分 
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++){
		if(a[i].p==1){
			int x=lower_bound(cnt+1,cnt+1+n,a[i].x)-cnt,
				y=lower_bound(cnt+1,cnt+1+n,a[i].y)-cnt;
			//因为数组是有序的,所以我们可以用二分得到离散化后的值
			merge(x,y); 
		}
	}
	for(int i=1;i<=m;i++){
		if(a[i].p==0){
			int x=lower_bound(cnt+1,cnt+1+n,a[i].x)-cnt,
				y=lower_bound(cnt+1,cnt+1+n,a[i].y)-cnt;
			//错误:lower_bound(cnt+1,cnt+1+d,a[i].y)-cnt; 去重过了 
			if(find(x)==find(y)){//互不相等却在同一集合内,自相矛盾 
				cout<<"NO"<<endl;
				return;
			}
		}
	}
	cout<<"YES"<<endl;
}
int main(){
	ios::sync_with_stdio(false);
	cin>>t;
	while(t--) solve();
	return 0;
}