并查集+离散化 acwing 237

题意：

有n个限制条件，i,j 1 表示xi和xj相等，i j 0 表示xi和xj不相等。如果出现了矛盾输出NO，没有出现矛盾输出YES

题解：

由于i 和j的数很大，所以需要离散化，把i和j都放在一个vector里面，然后对vec排序去重。

sort(vec.begin(), vec.end());
vec.erase(unique(vec.begin(), vec.end()), vec.end());

unique函数返回去重后的末尾，erase函数删除从起点到终点的元素。

先处理相同的元素的情况，就是直接把它们合并在一个集合里面。

然后再处理不同的情况，看是否出现冲突。

#pragma warning(disable:4996)
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<map>
#include<string>
#include<set> 
#include<queue>
using namespace std;
typedef long long ll;
vector<int>vec;
struct node
{
	int x, y, w;
	bool operator < (const node temp)
	{
		return w > temp.w;
	}
};
node a[1000005];
int fa[1000005];
int find(int x)
{
	if (x == fa[x])
		return x;
	else return fa[x] = find(fa[x]);
}
void merge(int x, int y)
{
	int fx, fy;
	fx = find(x); fy = find(y);
	fa[fx] = fy;
}
int main()
{
	int T, i, j, flag, n;
	scanf("%d", &T);
	while (T--)
	{
		scanf("%d", &n);
		vec.clear();
		for (i = 1; i <= 2 * n; i++)
		{
			fa[i] = i;
		}
		for (i = 1; i <= n; i++)
		{
			scanf("%d%d%d", &a[i].x, &a[i].y, &a[i].w);
			vec.push_back(a[i].x);
			vec.push_back(a[i].y);
		}
		sort(vec.begin(), vec.end());
		vec.erase(unique(vec.begin(), vec.end()), vec.end());
		sort(a + 1, a + 1 + n);
		flag = 1;
		int x, y;
		for (i = 1; i <= n; i++)
		{
			x = lower_bound(vec.begin(), vec.end(), a[i].x) - vec.begin() + 1;
			y = lower_bound(vec.begin(), vec.end(), a[i].y) - vec.begin() + 1;
			if (a[i].w)
			{
				merge(x, y);
			}
			else
			{
				int fx = find(x), fy = find(y);
				if (fx == fy)
				{
					flag = 0;
					break;
				}
			}
		}
		if (flag)printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}