算法竞赛进阶指南 0X40数据结构进阶——程序自动分析

题目链接

Acwing 237.程序自动分析 mid

题目描述

考虑一个约束满足问题的简化版本：假设 $x1,x2,x3,\dots$代表程序中出现的变量，给定 $n$ 个形如 $xi=xj$ 或 $xi\ne xj$ 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。

例如，一个问题中的约束条件为：$x1=x2，x2=x3，x3=x4，x1\ne x4$，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

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

输入格式

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

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

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

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

输出格式

输出文件包括 $t$ 行。

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

数据范围

• $1\le n\le 10^5$

• $1\le i,j\le 10^9$

输入样例：

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

输出样例：

NO
YES

解法：并查集

我们需要知道的是 $=$ 是有传递性的。 即 $a=b$ 并且 $b=c$ ，那么 $a=c$。

所以对于 等于关系 ，我们可以使用集合的方式来表示。比如：$a=b$ ，那么 $a,b$ 就在一个集合中，即 { a , b } \{a,b\} {a,b}；$b=c$，那么 $b,c$ 也在同一个集合中，即 { a , b , c } \{a,b,c\} {a,b,c}。

我们可以使用 并查集 来做到这一点。对于 $a=b$，那么我们就将 $a,b$ 加到一个集合中。

对于所有的 等于关系，我们都将其合并到一个集合中。最后遍历的时候，对于 不等于的关系，比如此时遍历到了 $x\ne y$，如果 $x$ 和 $y$ 在同一个集合之中，就说明输入的条件是不可能被满足的。因为在同一个集合中的 只能是等于关系。

注意： $i,j$ 的值域是 $10^9$，我们不可能开一个 $10^9$ 大小的数组。我们注意到 $i$ 和 $j$ 的数量最多有 $2n$ 个，所以我们可以将其离散化到 一个 $2n$ 的数组上。

时间复杂度： $O(n)$

C++代码：

#include<iostream>
#include<algorithm>
#include<map>

using namespace std;

const int N = 1e5+10 , M = 2 * N;
int f[M];

struct edge{
    int a,b,type;
}e[N];

map<int,int> mp;

int n , idx;

//离散化
int S(int x){
    if(mp.find(x) != mp.end()) return mp[x];
    return mp[x] = idx++;
}

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


void solve(){
    cin>>n;
    int t , a , b;
    
    idx = 1;
    mp.clear();
    for(int i = 1;i < M;i++) f[i] = i;
    for(int i = 1;i <= n;++i){
        scanf("%d%d%d",&a,&b,&t);
        //离散化
        a = S(a) , b = S(b);
        e[i] = {a,b,t};
        
        //t == 1 时 , 才是相等关系
        if(t){
            int x = find(a) , y = find(b);
            f[x] = y;
        }
    }
    
    bool ok = true;
    for(int i = 1;i <= n;i++){
        auto [a,b,t] = e[i];
        if(!t){
            int x = find(a) , y = find(b);
            if(x == y){
                ok = false;
                break;
            }
        }
    } 
    
    if(ok) puts("YES");
    else puts("NO");
}

int main(){
    int t;
    cin>>t;
    while(t--){
        solve();
    }
}