LeetCode打卡： 1466. 重新规划路线

n 座城市，从 0 到 n-1 编号，其间共有 n-1 条路线。因此，要想在两座不同城市之间旅行只有唯一一条路线可供选择（路线网形成一颗树）。去年，交通运输部决定重新规划路线，以改变交通拥堵的状况。

路线用 connections 表示，其中 connections[i] = [a, b] 表示从城市 a 到 b 的一条有向路线。

今年，城市 0 将会举办一场大型比赛，很多游客都想前往城市 0 。

请你帮助重新规划路线方向，使每个城市都可以访问城市 0 。返回需要变更方向的最小路线数。

题目数据 保证 每个城市在重新规划路线方向后都能到达城市 0 。

示例：

输入：n = 6, connections = [[0,1],[1,3],[2,3],[4,0],[4,5]]
输出：3
解释：更改以红色显示的路线的方向，使每个城市都可以到达城市 0 。

我的代码：

class Solution(object):
    def minReorder(self, n, connections):
        """
        :type n: int
        :type connections: List[List[int]]
        :rtype: int
        """
        # dfs
        # 这是一个树的问题，0是根节点
        # 用邻接表edge存储树的边，其中父节点指向子节点保存为正值（需要change），反之为负
        # 从根节点开始深度优先遍历，用visited数组记录遍历过的结点
        def dfs(edge, visited, start):
            res = 0
            visited[start] = True
            for e in edge[start]:
                if not visited[abs(e)]:
                    res += dfs(edge, visited, abs(e)) + (e > 0)
            return res
        
        visited = [False] * n
        edge = [[] for i in range(n)]
        for c in connections:
            edge[c[0]].append(c[1])
            edge[c[1]].append(-c[0])
            
        return dfs(edge, visited, 0)
        

Time： O(N)
Space: O(N)