Leetcode1466. 重新规划路线(medium,BFS)

目录

1. 问题描述

2. 解题分析

3. 代码实现


1. 问题描述

n 座城市,从 0 到 n-1 编号,其间共有 n-1 条路线。因此,要想在两座不同城市之间旅行只有唯一一条路线可供选择(路线网形成一颗树)。去年,交通运输部决定重新规划路线,以改变交通拥堵的状况。路线用 connections 表示,其中 connections[i] = [a, b] 表示从城市 a 到 b 的一条有向路线。

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

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

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

示例 1:

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

示例 2:

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

示例 3:

输入:n = 3, connections = [[1,0],[2,0]]
输出:0

提示:

  • 2 <= n <= 5 * 10^4
  • connections.length == n-1
  • connections[i].length == 2
  • 0 <= connections[i][0], connections[i][1] <= n-1
  • connections[i][0] != connections[i][1]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reorder-routes-to-make-all-paths-lead-to-the-city-zero
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题分析

        问题描述中的“路线网形成一颗树”是啥意思?从所举例来看,并不是树的样子啊。比如说,从示例1来看,节点2和节点4都是“根”。。。多个“根”还叫树吗?那不是应该叫森林吗?

        如果只是需要让每个节点都能到达节点0而翻转边的方向的话,(把图看作是一个无向图)从节点0出发以深度优先或者广度优先搜索的方式遍历每个节点,将需要反向的边执行反向操作即可。

        问题是如何证明这样即保证了反向边数最少呢?

        connections是edge列表,可以先基于edge列表建立邻接表以方便搜索。

        此外,由于需要反复查询搜索过程中所碰到的边的方向,而list的查询效率非常低,可以先将connections转变为tuple集合,以方便高效查询。 

        

3. 代码实现

import time
from typing import List
from collections import deque

class Solution:
    def minReorder(self, n: int, connections: List[List[int]]) -> int:
        
        adjlist = [[] for k in range(n)]
        edge_set = set()
        for edge in connections:
            adjlist[edge[0]].append(edge[1])
            adjlist[edge[1]].append(edge[0])
            edge_set.add(tuple(edge))
        
        q = deque([0])
        visited = set([0])
        revcnt  = 0
        while len(q)>0:
            city = q.popleft()
            for nxt in adjlist[city]:
                if nxt not in visited:
                    q.append(nxt)
                    visited.add(nxt)
                    # if [city, nxt] in connections:
                    if (city,nxt) in edge_set:
                        # This edge should be reversed.
                        revcnt += 1
        return revcnt
    
if __name__ == "__main__":

    sln = Solution()

    n = 6 
    connections = [[0,1],[1,3],[2,3],[4,0],[4,5]]    
    print(sln.minReorder(n, connections))
    
    n = 5
    connections = [[1,0],[1,2],[3,2],[3,4]]
    print(sln.minReorder(n, connections))
    
    n = 3
    connections = [[1,0],[2,0]]
    print(sln.minReorder(n, connections))

        执行用时:308 ms, 在所有 Python3 提交中击败了74.29%的用户

        内存消耗:36.1 MB, 在所有 Python3 提交中击败了63.43%的用户

        回到主目录:笨牛慢耕的Leetcode每日一解题解笔记(动态更新。。。) 

猜你喜欢

转载自blog.csdn.net/chenxy_bwave/article/details/124224649