题目描述
n 座城市,从 0 到 n-1 编号,其间共有 n-1 条路线。因此,要想在两座不同城市之间旅行只有唯一一条路线可供选择(路线网形成一颗树)。去年,交通运输部决定重新规划路线,以改变交通拥堵的状况。路线用 connections 表示,其中 connections[i] = [a, b] 表示从城市 a 到 b 的一条有向路线。今年,城市 0 将会举办一场大型比赛,很多游客都想前往城市 0 。请你帮助重新规划路线方向,使每个城市都可以访问城市 0 。返回需要变更方向的最小路线数。题目数据 保证 每个城市在重新规划路线方向后都能到达城市 0 。
解析
对图从0号城市开始遍历,如果不看方向,是连通图,因此是可以全部遍历的,那么如果是顺着遍历的,说明是从0往外指的,因此方向需要改变,否则不变。如果用邻接矩阵会超出内存,需要使用邻接表。
public int minReorder(int n, int[][] connections) {
List<int[]>[] e = new List[n];
for (int i = 0; i < n; i++) {
e[i] = new ArrayList<int[]>();
}
for (int[] edge : connections) {
e[edge[0]].add(new int[]{edge[1], 1});
e[edge[1]].add(new int[]{edge[0], 0});
}
return dfs(0, -1, e);
}
public int dfs(int x, int parent, List<int[]>[] e) {
int res = 0;
for (int[] edge : e[x]) {
if (edge[0] == parent) {
continue;
}
res += edge[1] + dfs(edge[0], x, e);
}
return res;
}