LeetCode-5426、重新規劃路線-中等
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]
代碼:
class Solution:
def minReorder(self, n: int, connections: List[List[int]]) -> int:
ans = 0
stack = set([0])
for a, b in connections:
if a in stack:
stack.add(b)
ans += 1
elif b in stack:
stack.add(a)
return ans