ac237程序自动分析

题目

扣我传送

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

考虑一个约束满足问题的简化版本：假设 x1,x2,x3,…

代表程序中出现的变量，给定 n 个形如 x_i=x_j 或 x_i≠x_j

的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。

例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4

，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

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

输入格式

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

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

第 1 行包含 1 个正整数 n，表示该问题中需要被满足的约束条件个数。

接下来 n 行，每行包括 3 个整数 i,j,e，描述 1 个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 e=1，则该约束条件为 x_i=x_j；若 e=0，则该约束条件为 x_i≠x_j。

输出格式

输出文件包括 t 行。

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

数据范围

1≤n≤10⁵

1≤i,j≤10⁹

输入样例：

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

输出样例：

NO
YES

---

思路以及代码

一道水题，思路很好想，但是如果不用stl的东西的话还是挺麻烦的。

思路

只需要考虑一次的判断后面加个循环就完事了。

并查集

相等是可传递的，我们可以把相等的放到同一个集合里，之后只需要判断在相等的基础上是否能够满足所有的不等即可

那很明显就是用并查集了

哈希一下

注意到里面 i，j 最大可到10⁹， 不好开这么大的数组。而且单个问题的条件最多也就10⁵个，所以我们需要哈希一下。

这里手搓就比较麻烦，stl里有一个叫map的容器很好用，但是放这里会超时。而我们不关心集合元素的顺序，所以用相近的unordered_map容器就可以了。

存不等关系之后处理

遇到不等关系的我们先存起来，最后再进行处理。这里用什么方式存都ok，我用的队列。

代码

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

int find_set(int i) {
    if (parent.count(i) == 0) return i;

    parent[i] = find_set(parent[i]);
    return parent[i];
}

int main()
{
    int repeat, conditions, x1, x2, state;
    queue<pair<int, int>> unequal;
    scanf("%d", &repeat);
    while (repeat--) {
        while (!unequal.empty()) unequal.pop();
        parent.clear();

        scanf("%d", &conditions);
        while (conditions--) {
            scanf("%d %d %d", &x1, &x2, &state);
            if (state == 0) 
                unequal.push({x1, x2});
            else if (find_set(x1) != find_set(x2)) 
                parent[find_set(x1)] = find_set(x2);
        }
        
        while (!unequal.empty()) {
            x1 = find_set(unequal.front().first);
            x2 = find_set(unequal.front().second);
            if (x1 == x2) break;
            unequal.pop();
        }

        if (unequal.empty()) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}