每日一题 1466. 重新规划路线（树，DFS）

eyvr

于 2023-12-07 13:20:28 发布

阅读量104

点赞数

CC 4.0 BY-SA版权

分类专栏：用Python刷力扣文章标签：深度优先算法 leetcode python

本文链接：https://blog.csdn.net/qq_46636391/article/details/134853082

用Python刷力扣专栏收录该内容

108 篇文章

订阅专栏

文章描述了一个名为Solution的类，其中包含一个minReorder函数，用于计算在给定无向树中从根节点0重新排列节点以使得每条边都按正确方向连接的最小操作次数。函数利用深度优先搜索（DFS）遍历树结构，避免重复路径。

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

在这里插入图片描述

根据 connections 建立无向树
从 0 开始深搜，每次调用 dfs 时判断路径方向是否正确

class Solution:
    def minReorder(self, n: int, connections: List[List[int]]) -> int:
        to = defaultdict(set)
        edge = defaultdict(list)
        for con in connections:
            edge[con[0]].append(con[1])
            edge[con[1]].append(con[0])
            to[con[0]].add(con[1])
        
        ans = 0
        def dfs(node, fa):
            nonlocal ans
            for ch in edge[node]:
                if ch == fa:
                    continue
                if ch in to[node]:
                    ans += 1
                dfs(ch, node)
        
        dfs(0, -1)
        return ans