第 191 场周赛 5426. 重新规划路线 DFS

5426. 重新规划路线 DFS

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]

因为是一颗有向树,所以我们可以重新建图,建成无向的,并且给每条边加上边权,

• 正向边就是正权,
• 反向边就是负权
• 从0号节点开始$dfs$对于每条负边权$ans++$

#define MAXN ((int)5e4+7)
struct Node {
    int to, dir; //dir是边权
} ;

int ans;
bool vis[MAXN];
vector<Node> G[MAXN];

void dfs(int u) {
    vis[u] = true;
    for(auto ed : G[u]) {
        int v = ed.to, w = ed.dir;
        if(vis[v]) continue ;
        if(w > 0) ans ++; //对于每个正权边 ans++
        // printf("%d %d w=%d\n", u, v, w);
        dfs(v);
    }
}

class Solution {
public:
    int minReorder(int n, vector<vector<int>>& vvi) {
        //初始化
        for(int i=0; i<=n+1; i++) G[i].clear(), vis[i] = false;
        for(auto v : vvi) //重新建图
            G[v[0]].push_back({v[1], 1}), /*正向边*/
      		G[v[1]].push_back({v[0], -1}); /*反向边*/
        ans = 0;
        dfs(0);
        
        return ans;
    }
};