C++刷leetcode(1466. 重新规划路线)【深度优先搜索】【有向图】
本帖最后由 糖逗 于 2020-6-7 14:12 编辑题目描述:
n 座城市,从 0 到 n-1 编号,其间共有 n-1 条路线。因此,要想在两座不同城市之间旅行只有唯一一条路线可供选择(路线网形成一颗树)。去年,交通运输部决定重新规划路线,以改变交通拥堵的状况。
路线用 connections 表示,其中 connections = 表示从城市 a 到 b 的一条有向路线。
今年,城市 0 将会举办一场大型比赛,很多游客都想前往城市 0 。
请你帮助重新规划路线方向,使每个城市都可以访问城市 0 。返回需要变更方向的最小路线数。
题目数据 保证 每个城市在重新规划路线方向后都能到达城市 0 。
示例 1:
输入:n = 6, connections = [,,,,]
输出:3
解释:更改以红色显示的路线的方向,使每个城市都可以到达城市 0 。
示例 2:
输入:n = 5, connections = [,,,]
输出:2
解释:更改以红色显示的路线的方向,使每个城市都可以到达城市 0 。
示例 3:
输入:n = 3, connections = [,]
输出:0
提示:
2 <= n <= 5 * 10^4
connections.length == n-1
connections.length == 2
0 <= connections, connections <= n-1
connections != connections
来源:力扣(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.size() == 1) return;
for(auto cha : store){
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].push_back(make_pair(cha, true));
store].push_back(make_pair(cha, false));
}
//深度优先搜索
int res = 0;
dfs(0, -1, store, res);
return res;
}
}; 参考评论区想法改写的{:10_327:}
页:
[1]