文章目录
- 1. 题目
- 2. 解题
- 2.1 DFS
- 2.2 BFS
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. 解题
建立无向图,在无向图上dfs 或者 bfs,从0开始
无向图上记录上是正向还是反向,遍历的时候,遇到反向的需要计数+1
2.1 DFS
class Solution {int count = 0;unordered_map<int,unordered_map<int, bool>> m;
public:int minReorder(int n, vector<vector<int>>& connections) {vector<bool> vis(n,false);for(auto& c : connections){m[c[0]][c[1]] = true;//等于true的需要反转m[c[1]][c[0]] = false;}vis[0] = true;dfs(0, vis);return count;}void dfs(int i, vector<bool> &vis){for(auto it = m[i].begin(); it != m[i].end(); ++it){if(!vis[it->first]){if(m[i][it->first])//是反向的count++;vis[it->first] = true;dfs(it->first, vis);}}}
};
960 ms 114.2 MB
2.2 BFS
class Solution {
public:int minReorder(int n, vector<vector<int>>& connections) {int count = 0, id;unordered_map<int,unordered_map<int, bool>> m;vector<bool> vis(n,false);for(auto& c : connections){m[c[0]][c[1]] = true;//等于true的需要反转m[c[1]][c[0]] = false;}vis[0] = true;queue<int> q;q.push(0);while(!q.empty()){id = q.front();q.pop();for(auto it = m[id].begin(); it != m[id].end(); ++it){if(!vis[it->first]){if(m[id][it->first])count++;vis[it->first] = true;q.push(it->first);}}}return count;}
};
984 ms 115.2 MB