leetcode：847. 访问所有节点的最短路径【bfs + 记忆化seen】

白速龙王的回眸

于 2022-08-15 13:49:40 发布

阅读量250

点赞数 1

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 宽度优先算法

本文链接：https://blog.csdn.net/weixin_40986490/article/details/126345366

leetcode 专栏收录该内容

781 篇文章

订阅专栏

博客围绕解决问题的算法展开。因n小采用状态压缩二进制，访问全部元素需mask全1，求最短路径用广度优先搜索（BFS），并用seen记录(now, mask)避免重复。遍历各点可作起点，记忆化(now, mask)，mask首次全1时记录最短路径。

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

在这里插入图片描述

分析

n这么小，天生的状态压缩二进制
然后要访问完全部所以mask要全1
最短路径的话要用bfs
然后用seen记录(now, mask)
再次遇到的话肯定不是最短的

ac code

class Solution:
    def shortestPathLength(self, graph: List[List[int]]) -> int:
        n = len(graph)


        # (now, mask)作为记忆化标签, cnt表示当前步数
        dq = deque([(i, 1 << i, 0) for i in range(n)])
        seen = set([(i, 1 << i) for i in range(n)])

        # bfs（最少的步数走完全部）
        while dq:
            now, mask, cnt = dq.popleft()
            if mask == (1 << n) - 1:
                return cnt
            for next in graph[now]:
                # 没有重复(now, mask)
                if (next, mask | (1 << next)) not in seen:
                    seen.add((next, mask | (1 << next)))
                    dq.append((next, mask | (1 << next), cnt + 1))