237. 程序自动分析 离散化 并查集

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

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

例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

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

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

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

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

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

输出格式
输出文件包括 t 行。

输出文件的第 k 行输出一个字符串 YES 或者 NO，YES 表示输入中的第 k 个问题判定为可以被满足，NO 表示不可被满足。

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5;
int p[N];
struct node{
	int x,y;
}q[N];
unordered_map<int,int>vis;
int get_fa(int x) {
	if (x == p[x])return p[x];
	return p[x] = get_fa(p[x]);
}
void merge(int x, int y) {
	int dx = get_fa(x);
	int dy = get_fa(y);
	if (dx != dy) {
		p[dx] = dy;
	}
}
int cnt = 0,tt=0;
int main() {
	int t;
	cin >> t;
	while (t--) {
		cnt=tt=0;
		vis.clear();
		for(int i=1;i<N;i++)
		{
			p[i]=i;
		}
		int n;
		cin>>n;
		for (int i = 1; i <= n; i++) {
			int u, v, e;
			cin >> u >> v >> e;
			if(vis[u]==0)
			{
				vis[u]=++cnt;
			}
			if(!vis[v])
			{
				vis[v]=++cnt;
			}
			if (e == 1) {
				merge(vis[u],vis[v]);
			}
			else
			{
				q[++tt].x=vis[u];
				q[tt].y=vis[v];
			}
		}
		int i;
		for(i=1;i<=tt;i++)
		{
			if(get_fa(q[i].y)==get_fa(q[i].x))
			{
				cout<<"NO"<<endl;
				break;
			}
		}
		if(i>tt)
		cout<<"YES"<<endl;
	}
	return 0;
}

因为x的范围太大所以需要离散处理一下。
map 与 unordered_map
一个使用红黑树实现，一个用哈希表实现。
map里面是具有有序性的，在需要排序的题里面好用，但是查找时间复杂度低于后者。