【并查集练习】NOI2015程序自动分析

 这题据dzm大佬的话是并查集板子题，然而卡了我很久，果然还是我太弱了吗……

 

---

 

思路：按秩合并和路径压缩在这道题都能用，对那些用两种优化如同写板子的OIer们算是很友好了。这道题总体思路是把条件相等的合并，条件不等的则判断是否已经在一个集合中，如果是则不成立。具体的来说，离线来做是个不错的选择。先将数据都读入，使e为1的在前，e为0的在后，先全部合并在判断，这样做对最后结果是没有影响的，该矛盾还是要矛盾。这样可以省去记录条件不等的i和j这件麻烦事。因为有些i和j非常大，我们可以在读入的时候模一个大质数（这样其实就可以A掉这道题，但不是最好的办法→离散化）。

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define clr(a) memset(a,0,sizeof(a))
using namespace std;
struct haha{LL x,y,e;}a[1000010];
LL n,m,mod=100000007;
LL fa[1000010],h[1000010],uni[1000010<<1];//uni是用来离散化的数组 
bool flag;
bool cmp(haha x,haha y){return x.e>y.e;}
LL get(LL x){return fa[x]==x?fa[x]:fa[x]=get(fa[x]);}//路径压缩 
void merge(LL x,LL y){//按秩合并 
	LL fx=get(x);
	LL fy=get(y);
	if(fx!=fy){
		if(h[fx]>=h[fy]){
			h[fx]+=h[fy];
			fa[fy]=fx;
		}
		else{
			h[fy]+=h[fx];
			fa[fx]=fy;
		}
	}
}
void first(LL k){//初始化 
	for(LL i=1;i<=k;i++){
		h[i]=1;
		fa[i]=i;
	}
}
int main(){
	scanf("%lld",&m);
	while(m--){
		LL cnt=0;
		clr(a);clr(uni);
		scanf("%lld",&n);
		for(LL i=1;i<=n;i++){
			scanf("%lld%lld%lld",&a[i].x,&a[i].y,&a[i].e);
			a[i].x%=mod;a[i].y%=mod;//避免过亿的数据，模大质数 
			uni[++cnt]=a[i].x;uni[++cnt]=a[i].y;
		}
		sort(uni+1,uni+1+cnt);//45~50：离散化，优化时空效率 
		LL len=unique(uni+1,uni+1+cnt)-uni-1;
		for(LL i=1;i<=n;i++){
			a[i].x=lower_bound(uni+1,uni+1+len,a[i].x)-uni-1;
			a[i].y=lower_bound(uni+1,uni+1+len,a[i].y)-uni-1;
		}
		first(len);
		flag=0;
		sort(a+1,a+1+n,cmp);
		for(LL i=1;i<=n;i++){
			LL xx=get(a[i].x);
			LL yy=get(a[i].y);
			if(a[i].e==1)
				merge(xx,yy);
			if(a[i].e==0)
				if(xx==yy){
					printf("NO\n");
					flag=1;
					break;
				}
		} 
		if(!flag) printf("YES\n");
	}
	return 0;
}

功德圆满