LeetCode 1466. 重新规划路线（DFS/BFS）

文章目录

- 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;}
};