題目鏈接

換根dp
首先dfs記錄每個節點到左右子節點的距離和sum，和子節點個數cnt。

已知父節點的(sum,cnt)如何向子節點轉移？
假設當前節點的信息為(sum,cnt)，父節點的信息為(Fsum,Fcnt)。
那么子節點的sum = sum+[Fsum-sum-(cnt+1)] + [n-(cnt+1)]

class Solution { public:vector<vector<int>> g; vector<int> sum, cnt, ans;int n;vector<int> sumOfDistancesInTree(int N, vector<vector<int>>& edges) {sum.resize(N, 0);cnt.resize(N, 0);ans.resize(N, 0);n = N;g.resize(N, vector<int>{});for (auto it : edges) {g[it[0]].push_back(it[1]);g[it[1]].push_back(it[0]);}dfs1(0, 0);dfs2(0, 0);ans[0] = sum[0];return ans;}void dfs2(int fa, int x) {for (auto it : g[x]) {if (it == fa) continue;ans[it] = sum[x] + n - 2*(cnt[it]+1);sum[it] = ans[it];dfs2(x, it);}}pair<int,int> dfs1(int fa, int x) {for (auto it : g[x]) {if (it == fa) continue;pair<int,int> ret = dfs1(x, it);cnt[x] += ret.second + 1;sum[x] += (ret.second+1) + ret.first;}return {sum[x], cnt[x]};} };

總結

以上是生活随笔為你收集整理的树形动态规划 - 树中距离之和的全部內容，希望文章能夠幫你解決所遇到的問題。

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