題目鏈接

換根dp
首先dfs記錄每個(gè)節(jié)點(diǎn)到左右子節(jié)點(diǎn)的距離和sum，和子節(jié)點(diǎn)個(gè)數(shù)cnt。

已知父節(jié)點(diǎn)的(sum,cnt)如何向子節(jié)點(diǎn)轉(zhuǎn)移？
假設(shè)當(dāng)前節(jié)點(diǎn)的信息為(sum,cnt)，父節(jié)點(diǎn)的信息為(Fsum,Fcnt)。
那么子節(jié)點(diǎn)的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]};} };

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。