python - 使用字典(python)中的特定键构建列表？

Question

我正在用 Python 实现 Dijkstra 搜索算法。在搜索结束时，我使用前驱 map 重建最短路径，从目标节点的前驱开始。例如:

path = []
path.append(destination)
previous = predecessor_map[destination]
while previous != origin:
    path.append(previous)
    previous = predecessor_map[previous]

有没有什么方法可以用更少的代码行来做到这一点(例如列表理解)？

Answer 1

我唯一的建议是摆脱轻微的代码重复:

path = []
previous = destination
while previous != origin:
    path.append(previous)
    previous = predecessor_map[previous]

除此之外，我认为您的代码实际上非常清晰，不太可能从任何缩短它的尝试中受益。

最后，值得注意的是，当 destination == origin 时，上述内容也有效，而您的原始版本很可能无效(取决于 predecessor_map 的准确程度人口稠密)。不知道这是否与您的用例相关。