LeetCode题解(1334)：阈值距离内邻居最少的城市(Python)

最新推荐文章于 2023-11-14 23:26:24 发布

原创最新推荐文章于 2023-11-14 23:26:24 发布 · 423 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #图 #广度优先搜索 #深度优先搜索

刷题记录专栏收录该内容

1966 篇文章

订阅专栏

本文介绍了一种解决城市间最短路径问题的算法，利用广度优先搜索（BFS）或深度优先搜索（DFS）来找到使得任意两城市间的最远距离不超过给定阈值的最小城市数。在给定的城市数量、边和距离阈值的情况下，通过构建图并遍历所有可能的起始城市，找到使经过的城市数量最少的那个城市作为答案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：原题链接（中等）

标签：图、广度优先搜索、深度优先搜索

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (N \times E)$	$O (N + E)$	912ms (37.33%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

class Solution:
    def findTheCity(self, n: int, edges: List[List[int]], distanceThreshold: int) -> int:
        graph = collections.defaultdict(dict)
        for edge in edges:
            graph[edge[0]][edge[1]] = edge[2]
            graph[edge[1]][edge[0]] = edge[2]

        ans_idx, ans_val = -1, n
        for city in range(n):
            visited = collections.Counter({city: 0})
            queue = collections.deque([(city, 0)])
            while queue:
                for _ in range(len(queue)):
                    city1, d1 = queue.popleft()
                    for city2 in graph[city1]:
                        d2 = d1 + graph[city1][city2]
                        if d2 <= distanceThreshold and (city2 not in visited or d2 < visited[city2]):
                            queue.append((city2, d2))
                            visited[city2] = d2
            if len(visited) <= ans_val:
                ans_idx, ans_val = city, len(visited)

        return ans_idx