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
解题思路
先标记那些边需要重新导向,那些边不需要重新导向,然后通过dfs对需要重新导向的边进行累加
class Solution {
public:
//x为当前正在访问的结点,parent为当前结点的父节点,e为已经标记好的边的集合
int dfs(int x,int parent,vector<vector<pair<int,int>>>& e){
int res=0;//有几条边需要被导向
for(auto &edge:e[x]){
//edge.first: 表示与当前节点x相连的另一个节点
if(edge.first==parent){//如果另一个结点为当前结点的父结点,则跳过,避免重复访问
continue;
}
//edge.second是一个标记,表示这条边是否需要被重新导向,为0不需要,为1需要
res+=edge.second+dfs(edge.first,x,e);
}
return res;
}
int minReorder(int n, vector<vector<int>>& connections) {
vector<vector<pair<int,int>>> e(n);
for(auto edge:connections){
//edge[0]表示为某条边的起始点,edge[1]表示为某条边的终点
//先标记那些边需要重新导向,那些边不需要
//下面表示为,edge[0]到edge[1]的边是否需要被导向,1需要,0不需要
e[edge[0]].push_back(make_pair(edge[1],1));
e[edge[1]].push_back(make_pair(edge[0],0));
}
return dfs(0,-1,e);
}
};