Daimayuan 501 距离和（换根树形dp 入门讲解）

Brightess

已于 2022-08-21 22:31:03 修改

阅读量245

点赞数

CC 4.0 BY-SA版权

分类专栏：树形dp 深搜文章标签：算法数据结构 c++

于 2022-08-21 22:22:53 首次发布

本文链接：https://blog.csdn.net/Jacob0824/article/details/126453021

深搜同时被 2 个专栏收录

31 篇文章

订阅专栏

树形dp

15 篇文章

订阅专栏

本文解析了如何使用树形动态规划（Dynamic Programming）解决例题Daimayuan501中关于树的节点为根时节点间距离之和的问题。通过两次扫描，首次计算以1为根的总和，然后利用换根状态方程进行递归更新，最终求得所有节点作为根的总距离。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树形dp 换根dp讲解（来自 OI Wiki）

在这里插入图片描述

例题 Daimayuan 501 距离和

在这里插入图片描述

题意：

给定 一个 n 个点的树，请求出这棵树 以各个节点为根时，除了根以外的点 到根节点的 距离之和。

思路：

二次扫描。

第一次扫描时，求出 以 1 为根节点时的答案。

第二次扫描时，依照推导而来的 换根dp 状态方程，求出 以每个点为根节点时的答案。

代码：

#include <bits/stdc++.h>

using namespace std;
//#define map unordered_map
#define int long long
const int N = 1e5 + 10;
vector<int> g[N];
int n;
int sz[N];
int dp[N];
int depth[N];

int dfs1(int u, int f)
{
	sz[u] = 1;
	depth[u] = depth[f] + 1;
	if (u == 1) depth[u] = 0;
	for (auto v : g[u])
	{
		if (v == f) continue;
		sz[u] += dfs1(v, u);
	}
	return sz[u];
}

void dfs2(int u, int f)
{
	for (auto v : g[u])
	{
		if (v == f) continue;
		dp[v] = dp[u] + n - 2 * sz[v];
		dfs2(v, u);
	}
}

signed main()
{
	cin >> n;
	int t = n - 1;
	while (t--)
	{
		int x, y; scanf("%lld%lld", &x, &y);
		g[x].emplace_back(y);
		g[y].emplace_back(x);
	}
	dfs1(1, 0);
	for (int i = 1; i <= n; ++i)
	{
		dp[1] += depth[i];
	}
	dfs2(1, 0);
	for (int i = 1; i <= n; ++i)
	{
		printf("%lld\n", dp[i]);
	}

	return 0;
}