洛谷刷题C++语言 | P1955 程序自动分析

​学习C++从娃娃抓起！记录下洛谷C++学习和备考过程中的题目，记录每一个瞬间。

附上汇总贴：洛谷刷题C++语言 | 汇总

---

【题目描述】

在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本：假设 $x_1,x_2,x_3,\dots$ 代表程序中出现的变量，给定 n 个形如 $x_i=x_j$ 或 $x_i\ne x_j$ 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：$x_1=x_2,x_2=x_3,x_3=x_4,x_4\ne x_1$，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

【输入】

输入的第一行包含一个正整数 t，表示需要判定的问题个数。注意这些问题之间是相互独立的。

对于每个问题，包含若干行：

第一行包含一个正整数 n，表示该问题中需要被满足的约束条件个数。接下来 n 行，每行包括三个整数 i,j,e，描述一个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 e=1，则该约束条件为 x_i=x_j。若e=0，则该约束条件为 $x_i\ne x_j$。

【输出】

输出包括 t 行。

输出文件的第 k 行输出一个字符串 YES 或者 NO（字母全部大写），YES 表示输入中的第 k 个问题判定为可以被满足，NO 表示不可被满足。

【输入样例】

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

【输出样例】

NO
YES

【代码详解】

#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
int t, n, cnt;
int a[N], b[N], e[N], p[N*2];
unordered_map<int, int> mp;
int find(int x)
{
    return p[x]==x ? x : p[x] = find(p[x]);
}
int get(int x)  //离散化
{
    if (mp.count(x)==0) mp[x] = ++cnt;
    return mp[x];
}
int main()
{
    cin >> t;
    while (t--) {
        cnt = 0;
        mp.clear();  //mp初始化
        cin >> n;
        for (int i=1; i<=n*2; i++) p[i] = i;
        for (int i=1; i<=n; i++) {
            cin >> a[i] >> b[i] >> e[i];
            a[i] = get(a[i]);
            b[i] = get(b[i]);
        }
        for (int i=1; i<=n; i++) {  //合并集合
            if (e[i]==1) {
                int x = find(a[i]), y = find(b[i]);
                if (x!=y) p[x] = y;
            }
        }
        bool flag = 1;
        for (int i=1; i<=n; i++) {  //判断矛盾
            if (e[i]==0 && find(a[i])==find(b[i])) {
                flag = 0;
                break;
            }
        }
        flag ? puts("YES") : puts("NO");
    }
    return 0;
}

【运行结果】

2
2
1 2 1
1 2 0
NO
2
1 2 1
2 1 1
YES