Leetcode 3532. Path Existence Queries in a Graph I

• Leetcode 3532. Path Existence Queries in a Graph I
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3532. Path Existence Queries in a Graph I

1. 解题思路

这一题算是一个比较典型的DSU的题目，我们就是不断地根据前后节点的距离将其进行聚类，然后考察query当中的两个节点是否属于同一个簇即可。

而关于DSU，它是非常经典的算法了，网上到处都是，我自己也写过一篇博客（《经典算法：并查集（DSU）结构简介》）作为备忘，因此这里就不过多展开了。

2. 代码实现

给出python代码实现如下：

class DSU:
    def __init__(self, N):
        self.root = [i for i in range(N)]
        
    def find(self, k):
        if self.root[k] != k:
            self.root[k] = self.find(self.root[k])
        return self.root[k]
    
    def union(self, a, b):
        x = self.find(a)
        y = self.find(b)
        if x != y:
            self.root[y] = x
        return

class Solution:
    def pathExistenceQueries(self, n: int, nums: List[int], maxDiff: int, queries: List[List[int]]) -> List[bool]:
        dsu = DSU(n)
        for i in range(n-1):
            if nums[i+1] - nums[i] <= maxDiff:
                dsu.union(i, i+1)
        
        return [dsu.find(u) == dsu.find(v) for u, v in queries]

提交代码评测得到：耗时309ms，占用内存49.6MB。