- 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] = xreturnclass 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。