LeetCode 1466. 重新规划路线（DFS/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
举报
腾讯云开发者
