Leetcode 1466. Reorder Routes to Make All Paths Lead to the City Zero (python)

最新推荐文章于 2023-12-07 10:40:53 发布

努利！奋斗！

最新推荐文章于 2023-12-07 10:40:53 发布

阅读量303

点赞数

本文链接：https://blog.csdn.net/qq_37821701/article/details/109772176

版权

该博客介绍了一道LeetCode上的题目，讨论如何通过BFS或DFS遍历一棵以城市0为根的树，调整边的方向以确保所有路径都指向城市0。博主提供了BFS和DFS两种解法，通过遍历节点和边，检查并调整边的顺序以满足条件。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

解法：

这道题目有个关键是，图里面n个节点，有n-1条边，而且题目也提示了可以看作是一棵树。关键在于理解每个节点都指向0代表什么：代表的是每个以0作为根节点，每个叶节点到根节点的路径都是子节点指向父节点的。比如第一个例子，只需要保证2到0和5到0这两条路径都是子节点指向父节点的。而对于connections里面的边，一定是有一个在树里是父节点，另一个是子节点，我们只需要保证这两个节点的相对顺序即可，那么解法就很清晰了：

从0节点出发，利用BFS或者DFS遍历一遍树，并且保存每个节点被访问的相对顺序
遍历一遍边，根据之前保存的访问的相对顺序是不是符合边的指向关系，如果不符合，表明这条边需要反转

当然也可以通过保存每个结点的父节点来判断相对关系，但是相对来说上面这种方法更加方便

BFS版本

class Solution:
    def minReorder(self, n: int, connections: List[List[int]]) -> int:
        tree = collections.defaultdict(list)
        # create graph to store the adjacent nodes
        for edge in connections:
            tree[edge[0]].append(edge[1])
            tree