C++刷leetcode（1466. 重新规划路线）【深度优先搜索】【有向图】

糖逗

题目描述：

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    void dfs(int node, int parent, map<int, vector<pair<int, bool> > >& store, int& res){
        if(node != 0 && store[node].size() == 1) return;
        for(auto cha : store[node]){
            if(cha.first != parent){//去除自环的情况
                if(cha.second == true) res++;
                cout  << cha.first << endl;
                dfs(cha.first, node, store, res);
            }
        }
    }
    int minReorder(int n, vector<vector<int>>& connections) {
        //在建立无向邻接信息的基础上添加指向的bool型数据，构建邻接信息
        map<int, vector<pair<int, bool> > > store;
        for(auto cha : connections){
            store[cha[0]].push_back(make_pair(cha[1], true));
            store[cha[1]].push_back(make_pair(cha[0], false));
        }
        //深度优先搜索
        int res = 0;
        dfs(0, -1, store, res);
        return res;
    }
};

糖逗

参考评论区想法改写的