LeetCode 834. 树中距离之和（树上DP）*

文章目录

1. 题目
2. 解题

1. 题目

给定一个无向、连通的树。树中有 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-of-distances-in-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

抄的官方答案

class Solution {vector<int> ans, son, dp;vector<vector<int>> g;
public:vector<int> sumOfDistancesInTree(int N, vector<vector<int>>& edges) {ans.resize(N);son.resize(N);dp.resize(N);g.resize(N);for(auto& e : edges){g[e[0]].push_back(e[1]);g[e[1]].push_back(e[0]);}dfs1(0, -1);//dp 初始化dfs2(0, -1);//dp 递推return ans;}void dfs1(int u, int f){    //求子树包含的节点数量// 第一次dfs 得到的每个节点子树节点到该节点的距离和son[u] = 1;dp[u] = 0;for(auto v : g[u]){if(v == f)continue;dfs1(v, u);dp[u] += dp[v]+son[v];//子节点的距离和 + 边被占用次数son[u] += son[v];}}void dfs2(int u, int f){ans[u] = dp[u];for(auto v : g[u]){if(v == f)continue;int dpu = dp[u], dpv = dp[v];//记录状态int sonu = son[u], sonv = son[v];dp[u] -= dp[v]+son[v];//跟换根节点为 v ，u 变成子节点son[u] -= son[v];dp[v] += dp[u]+son[u];// v 加上 u 的数据son[v] += son[u];dfs2(v, u);dp[u] = dpu, dp[v] = dpv;//回溯状态son[u] = sonu, son[v] = sonv;}}
};

148 ms 26.4 MB

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！

LeetCode 834. 树中距离之和（树上DP）*相关推荐

leetcode 834. 树中距离之和（dp）
给定一个无向.连通的树.树中有 N 个标记为 0...N-1 的节点以及 N-1 条边 .第 i 条边连接节点 edges[i][0] 和 edges[i][1] .返回一个表示节点 i 与其他所有节 ...
834. 树中距离之和
834. 树中距离之和解法一:超时的暴力解法 #include <iostream> #include <vector> #include <string> #i ...
834 树中距离之和
这道题我自己的想法只有对每个点都用一遍Dijkstra然后再求和,显然会超时,所以我都没有尝试. 研究了一下题解,发现题解很巧妙,自己对树的处理还是太稚嫩,之前树链剖分学的都忘光了. 对于固定根节点的 ...
leetcode 834. Sum of Distances in Tree | 834. 树中距离之和（树形DP）
题目 https://leetcode.com/problems/sum-of-distances-in-tree/ 题解一般的算法题,指令条数为 10^8 以内是可以通过的.也就是说,如果 arr ...
树形动态规划 - 树中距离之和
题目链接换根dp 首先dfs记录每个节点到左右子节点的距离和sum,和子节点个数cnt. 已知父节点的(sum,cnt)如何向子节点转移? 假设当前节点的信息为(sum,cnt),父节点的信息为(F ...
2019长安大学ACM校赛网络同步赛C LaTale （树上DP）
链接:https://ac.nowcoder.com/acm/contest/897/C 来源:牛客网 LaTale 时间限制:C/C++ 2秒,其他语言4秒空间限制:C/C++ 32768K,其他 ...
算法提高——树上DP（树的直径）
文章目录前言一.什么是树的直径? 二.例题三.树上DP 总结前言树的直径是图论里边非常高的考察点并且是入门树形dp的基础,竞赛的同学务必重视. 一.什么是树的直径? 树上最远两点(叶子结点) ...
【ZJOI2019】线段树【线段树上dp】【大讨论】
题意:有一个 [1,n][1,n][1,n] 的线段树和 mmm 个区间赋值操作,求任取一个操作的子集并按顺序在线段树上跑后线段树上有 lazy 标记的点的个数之和模 998244353998244 ...
牛客网CSP-S提高组赛前集训营1题解（仓鼠的石子游戏 [博弈论] + 乃爱与城市的拥挤程度 [树上DP] + 小w的魔术扑克[dfs + 离线]）
文章目录 T1:仓鼠的石子游戏题目题解代码实现 T2:乃爱与城市拥挤程度题目题解代码实现 T3:小w的魔术扑克题目题解代码实现 T1:仓鼠的石子游戏题目仓鼠和兔子被禁止玩电脑,无 ...

LeetCode 834. 树中距离之和（树上DP）*

文章目录

1. 题目

2. 解题

LeetCode 834. 树中距离之和（树上DP）*相关推荐

最新文章

热门文章