题面:
n 座城市,从 0 到 n-1 编号,其间共有 n-1 条路线。因此,要想在两座不同城市之间旅行只有唯一一条路线可供选择(路线网形成一颗树)。去年,交通运输部决定重新规划路线,以改变交通拥堵的状况。
路线用 connections 表示,其中 connections[i] = [a, b] 表示从城市 a 到 b 的一条有向路线。
今年,城市 0 将会举办一场大型比赛,很多游客都想前往城市 0 。
请你帮助重新规划路线方向,使每个城市都可以访问城市 0 。返回需要变更方向的最小路线数。
题目数据 保证 每个城市在重新规划路线方向后都能到达城市 0 。
示例1:
解法:
因为给定的图是一棵树,从结果开始想,最后每个点到达点0一定只有一种路线,最后每条边的方向是确定的。
可以将给定的有向边存为无向边,边上打个方向标记,然后从点0开始bfs扩展,遇到需要反转的边就累加答案。
并查集什么的也可以实现,主要思维点在于从点0开始进行无向图拓展
code:
class Solution {
public:
vector<pair<int,int> >g[50000+5];
int mark[50000+5];
int minReorder(int n, vector<vector<int>>& cc) {
for(auto i:cc){
int x=i[0],y=i[1];
g[x].push_back({y,0});//正向
g[y].push_back({x,1});//反向
}
for(int i=0;i<n;i++)mark[i]=0;
queue<int>q;
q.push(0);
mark[0]=1;
int ans=0;
while(!q.empty()){
int x=q.front();q.pop();
for(auto i:g[x]){
int v=i.first,flag=i.second;
if(!mark[v]){
mark[v]=1;
q.push(v);
if(flag==0){//正向边对答案有贡献
ans++;
}
}
}
}
return ans;
}
};