Java实现 LeetCode 834 树中距离之和（DFS+分析）

最新推荐文章于 2023-01-05 21:02:01 发布

南墙

最新推荐文章于 2023-01-05 21:02:01 发布

阅读量9.5k

点赞数 16

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： Java 树中距离之和 DFS 分析

本文链接：https://blog.csdn.net/a1439775520/article/details/106044013

834. 树中距离之和

给定一个无向、连通的树。树中有 N 个标记为 0…N-1 的节点以及 N-1 条边。

第 i 条边连接节点 edges[i][0] 和 edges[i][1] 。

返回一个表示节点 i 与其他所有节点距离之和的列表 ans。

示例 1:

输入: N = 6, edges = [[0,1],[0,2],[2,3],[2,4],[2,5]]
输出: [8,12,6,10,10,10]
解释:
如下为给定的树的示意图：
0
/
1 2
/|
3 4 5

我们可以计算出 dist(0,1) + dist(0,2) + dist(0,3) + dist(0,4) + dist(0,5)
也就是 1 + 1 + 2 + 2 + 2 = 8。因此，answer[0] = 8，以此类推。
说明: 1 <= N <= 10000

PS：
这个题看了半个小时没找到思路，只能去看题解，原来我的菜

class Solution {
   
   
    //这个题我看了半天，怎么看怎么超时，原来是中间有规律，该说不说，这个规律真的难找
  int[] ans, count;
    List<Set<Integer>> graph;
    int N;
    public int[] sumOfDistancesInTree(int N, int[][] edges) {
   
   
        this.N = N;
        graph = new