LeetCode刷题总结 1627 带阈值的图连通性

---

题目

有 n 座城市，编号从 1 到 n 。编号为 x 和 y 的两座城市直接连通的前提是： x 和 y 的公因数中，至少有一个 严格大于 某个阈值 threshold 。更正式地说，如果存在整数 z ，且满足以下所有条件，则编号 x 和 y 的城市之间有一条道路：

x % z == 0
y % z == 0
z > threshold
给你两个整数 n 和 threshold ，以及一个待查询数组，请你判断每个查询 queries[i] = [ai, bi] 指向的城市 ai 和 bi 是否连通（即，它们之间是否存在一条路径）。

返回数组 answer ，其中answer.length == queries.length 。如果第 i 个查询中指向的城市 ai 和 bi 连通，则 answer[i] 为 true ；如果不连通，则 answer[i] 为 false 。

示例 1：
输入：n = 6, threshold = 2, queries = [[1,4],[2,5],[3,6]]
输出：[false,false,true]
解释：每个数的因数如下：
1: 1
2: 1, 2
3: 1, 3
4: 1, 2, 4
5: 1, 5
6: 1, 2, 3, 6
所有大于阈值的的因数已经加粗标识，只有城市 3 和 6 共享公约数 3 ，因此结果是：
[1,4] 1 与 4 不连通
[2,5] 2 与 5 不连通
[3,6] 3 与 6 连通，存在路径 3–6

示例 2：

输入：n = 6, threshold = 0, queries = [[4,5],[3,4],[3,2],[2,6],[1,3]]
输出：[true,true,true,true,true]
解释：每个数的因数与上一个例子相同。但是，由于阈值为 0 ，所有的因数都大于阈值。因为所有的数字共享公因数 1 ，所以所有的城市都互相连通。
示例 3：

输入：n = 5, threshold = 1, queries = [[4,5],[4,5],[3,2],[2,3],[3,4]]
输出：[false,false,false,false,false]
解释：只有城市 2 和 4 共享的公约数 2 严格大于阈值 1 ，所以只有这两座城市是连通的。
注意，同一对节点 [x, y] 可以有多个查询，并且查询 [x，y] 等同于查询 [y，x] 。

原题链接: link.

---

思路

解题思路：
题目意思是，对于如果两个数连通，即存在一个大于阈值的公因数。

本题是要判断连通性，对于给定n个数，利用图的数据结构来解题。给满足两个数含有大于阈值的公因子连边。本题需要为两个功能：建图和查询

建图：
主要是考虑如何连边的问题：
假设P是A、B的公因数，假设P是A和B的一个公因数，那么显然P也是P和A的公因数，同时也是P和B的公因数。因此，我们就可以不连接(A,B)(A,B)，而是连接(P,A)(P,A)和(P,B)(P,B)。也就是说，我们可以枚举因子，然后将其所有的倍数与这一因子之间连边。

然后考虑阈值
要使得因子大于阈值，只需从(阈值+1)开始枚举因子即可，枚举其倍数，联通它和它的倍数节点

查询：
对于连通性，使用并查集来实现
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
通过比较根节点就知道是不是属于同一颗树（同一个集合），如果是，那么就不需要进行额外的操作；如果不是，就将两个树合为一个树。

---

代码（带详细注释）

#include <vector>
#include <stdlib.h>
#include <numeric>
using namespace std;

class UF {
public:
    vector<int> fa;     //根节点集
    vector<int> sz;     //记录连接节点个数
    int n;
    int comp_cnt;
    
//初始化，其中fa[]初始化为下标值，sz初始化为1
public:
    UF(int _n): n(_n), comp_cnt(_n), fa(_n), sz(_n, 1) {
        iota(fa.begin(), fa.end(), 0);
    }
    
    //并查集
    //用以判断两个点是否在同一个棵树上：查询其根节点，比较根节点是否相等
    int findset(int x) {
        return fa[x] == x ? x : fa[x] = findset(fa[x]);
    }
    
    //合并点
    void unite(int x, int y) {
        x = findset(x);
        y = findset(y);
        //x!=y，则xy不在同一棵树上，合并
        if (x != y) {
            //选择sz[]更大的点，即将顶点数小的连到顶点数大的上
            if (sz[x] < sz[y]) {
                swap(x, y);
            }
            fa[y] = x;
            sz[x] += sz[y];
            --comp_cnt;
        }
    }
    
    //通过查询根节点判断是否在一棵树上，即是否连通
    bool connected(int x, int y) {
        x = findset(x);
        y = findset(y);
        return x == y;
    }
};

class Solution {
public:
    vector<bool> areConnected(int n, int threshold, vector<vector<int>>& queries) {
        UF uf(n + 1);
        // 枚举公因数
        for (int z = threshold + 1; z <= n; ++z) {
            // 枚举两个 z 的倍数的点并连接
            for (int p = z, q = z * 2; q <= n; p += z, q += z) {
                uf.unite(p, q);
            }
        }
        
        vector<bool> ans;
        for (const auto& q: queries) {
            int x = q[0];
            int y = q[1];
            ans.push_back(uf.connected(x, y));
        }
        return ans;
    }
};

---

注释：
参考链接
作者：zerotrac2链接: link.
来源：力扣（LeetCode）

作者：lucifer1004link.
来源：力扣（LeetCode）