程序自动分析 （不保序离散化+并查集）

在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。考虑一个约束满足问题的简化版本：假设 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 表示不可被满足。

数据范围

1≤n≤10e5     1≤i,j≤10e9

输入样例：

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出样例：

NO
YES

思路：

由于数据个数少，但范围较大，所以我们首先要用离散化映射一下。

离散化：离散化（保序）_星河边采花的博客-CSDN博客

因为需要判定的问题是无先后顺序的，我们应该先把相等关系加到某个并查集里，然后判断不相等关系时，正好对应并查集查询操作。

代码：

#include <bits/stdc++.h>
#include <unordered_map> 
using namespace std;

const int N = 200010;
int n, m;
int p[N];
unordered_map<int, int> S;
struct Query {
    int x, y, e;
}query[N];

int get(int x) {
    if (S.count(x) == 0)S[x] = ++n;
    return S[x];
}

int find(int x) {
    if (p[x] != x)p[x] = find(p[x]);
    return p[x];
}

int main()
{
    int T;
    scanf("%d", &T);
    while (T--) {
        n = 0;
        S.clear();
        scanf("%d", &m);
        for (int i = 0; i < m; i++) {
            int x, y, e;
            scanf("%d%d%d", &x, &y, &e);
            query[i] = { get(x),get(y),e };
        }
        for (int i = 0; i <= n; i++) p[i] = i;

        for (int i = 0; i < m; i++) {
            if (query[i].e == 1) {
                int fa = find(query[i].x), fb = find(query[i].y);
                p[fa] = fb;
            }
        }
        bool fg = true;
        for (int i = 0; i < m; i++) {
            if (query[i].e == 0) {
                int pa = find(query[i].x), pb = find(query[i].y);
                if (pa == pb) {
                    fg = false;
                    break;
                }
            }
        }
        if (fg)cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}