程序自动分析（并查集，离散化）

程序自动分析（并查集，离散化）

n次给定两个正整数 $i,j$，表示两个整数变量 $x_i$ 与 $x_j$ 是相等的或者是不等的，问这些变量取一定的值能不能满足上述条件。

链接

---

分析

1. 这些关系的约数条件出现的顺序对结果没有影响
2. 只考虑相等关系是不会出现矛盾的
3. 考虑完所有相等的条件后，矛盾出现当且仅当给的条件是相等的两个变量不等

本题 $i,j<10^9$，但是 $n<10^5$，所以要离散化

代码

#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;
}