题意:
有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;
}