leetcode做题笔记1466. 重新规划路线

si_mple_2023-12-11 22:24

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

思路一：DFS

c++解法

cpp 复制代码

class Solution {
public:
    int DFS(int x,int parent,vector<vector<pair<int,int>>> &e){
        int res = 0;
        for (auto &edge : e[x]) {
            if (edge.first == parent) {
                continue;
            }
            res += edge.second + DFS(edge.first, x, e);
        }
            return res; 
        }
 int minReorder(int n, vector<vector<int>>& connections) {
    vector<vector<pair<int, int>>> e(n);
    for (int i = 0; i < connections.size(); i++) { 
        int from = connections[i][0]; 
        int to = connections[i][1]; 
        e[from].push_back(make_pair(to, 1)); 
        e[to].push_back(make_pair(from, 0));
    }
    return DFS(0, -1, e); 
}

};