給定一個無向、連通的樹。樹中有 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

代碼

class Solution {int[] sz,dp,re;Map<Integer,List<Integer>> map3=new HashMap<>();public int[] sumOfDistancesInTree(int N, int[][] edges) {re=new int[N];sz=new int[N];dp=new int[N];for (int i=0;i<N;i++)map3.put(i,new ArrayList<>());for(int[] edge:edges)//鄰接表{map3.get(edge[0]).add(edge[1]);map3.get(edge[1]).add(edge[0]);}getSumOfDistancesInTree(0, -1);getSumOfDistancesInTree2(0, -1);return re;}public void getSumOfDistancesInTree(int cur, int father) {//從根節點向下統計距離dp[cur]=0;sz[cur]=1;for(int next:map3.get(cur)){if(next==father) continue;getSumOfDistancesInTree(next,cur);dp[cur]+=dp[next]+sz[next];sz[cur]+=sz[next];}}public void getSumOfDistancesInTree2(int cur, int father) {re[cur]=dp[cur];for(int next:map3.get(cur)){if(next==father) continue;int pn=dp[next],pc=dp[cur],sn=sz[next],sc=sz[cur];dp[cur]-=dp[next]+sz[next];sz[cur]-=sz[next];dp[next]+=dp[cur]+sz[cur];sz[next]+=sz[cur];//將父子節點顛倒，對應的值發生改變getSumOfDistancesInTree2(next,cur);dp[next]=pn;dp[cur]=pc;sz[cur]=sc;sz[next]=sn;//還原父子節點}} }

總結

以上是生活随笔為你收集整理的leetcode 834. 树中距离之和（dp）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。