acwing 237 程序自动分析（并查集）

题目地址

用 并查集将所有相关的(1)都合并到相关集合里面，再去判断所有不想关的(0)是否出现在同一个集合，若出现在同一个集合内的话说明矛盾。

要用离散化去离散i,j再用find去查找i,j离散后的位置。

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
 
const int INF = 0x3f3f3f3f;
const double eps = 1e-4;
const int mod = 999911659;
const int N = 1e6+10;

int fa[N];
vector<int> se;
vector<pii> ve1,ve0;

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

void init(int n){
	ve0.clear();
	ve1.clear();
	se.clear();
	for(int i=0;i<=n;i++) fa[i] = i;
}

int find(int x){
	int t = lower_bound(se.begin(),se.end(),x)-se.begin();
	return t;
}
signed main(){
	IOS;
	#ifdef ddgo
		freopen("C:/Users/asus/Desktop/ddgoin.txt","r",stdin);
	#endif
	
	int tt; cin>>tt;
	while(tt --){
		int n; cin>>n;
		init(2*n);
		for(int i=0;i<n;i++){
			int a,b,c; cin>>a>>b>>c;
			se.push_back(a);
			se.push_back(b);
			if(c == 1) ve1.push_back({a,b});
			else ve0.push_back({a,b});
		}
		bool plas = true;
		sort(se.begin(),se.end());
		se.erase(unique(se.begin(),se.end()),se.end());
		for(auto i:ve1){
			int a = find(i.first),b = find(i.second);
			fa[get(a)] = get(b);
		}
		for(auto i:ve0){
			int a = find(i.first),b = find(i.second);
			if(get(a) == get(b)){
				plas = false;
				break;
			}
		}
		cout<<(plas?"YES":"NO")<<endl;
	}
	
    return 0;
}