程序自动分析(并查集,离散化)
n次给定两个正整数 i , j i,j i,j,表示两个整数变量 x i x_i xi 与 x j x_j xj 是相等的或者是不等的,问这些变量取一定的值能不能满足上述条件。
分析
- 这些关系的约数条件出现的顺序对结果没有影响
- 只考虑相等关系是不会出现矛盾的
- 考虑完所有相等的条件后,矛盾出现当且仅当给的条件是相等的两个变量不等
本题 i , j < 1 0 9 i,j<10^9 i,j<109,但是 n < 1 0 5 n < 10^5 n<105,所以要离散化
代码
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int n;
int fa[N << 1], raw[N << 1];
struct node
{
int x, y, z;
}a[N];
int get(int x)
{
if (fa[x] == x) return x;
return fa[x] = get(fa[x]);
}
int main()
{
int t;
scanf("%d", &t);
while (t--)
{
scanf("%d", &n);
for (int i = 1; i <= n; i++)
{
scanf("%d%d%d", &a[i].x, &a[i].y, &a[i].z);
raw[(i << 1) - 1] = a[i].x;
raw[i << 1] = a[i].y;
}
sort(raw + 1, raw + 2 * n + 1);
int m = unique(raw + 1, raw + 2 * n + 1) - raw - 1;
for (int i = 1; i <= m; i++) fa[i] = i;
for (int i = 1; i <= n; i++)
{
a[i].x = lower_bound(raw + 1, raw + m + 1, a[i].x) - raw;
a[i].y = lower_bound(raw + 1, raw + m + 1, a[i].y) - raw;
}
for (int i = 1; i <= n; i++)
if (a[i].z)
fa[get(a[i].x)] = get(a[i].y);
int res = 1;
for (int i = 1; i <= n; i++)
if (!a[i].z && get(a[i].x) == get(a[i].y))
{
res = 0;
break;
}
if (res) puts("YES");
else puts("NO");
}
return 0;
}