CSES-1133 Tree Distances II-CSDN博客

本文链接：https://blog.csdn.net/2403_87021226/article/details/144773835

题目传送门https://vjudge.net/problem/CSES-1133#author=GPT_zh

解题思路

先求出根节点 1 号点的答案，和每个点的子树大小 $sz_i$

设 $ans_i$ 为 $i$ 号点的答案。

对于每一个点 $i$ 和它的根节点 $fa$ ， $ans_i$ 和 $ans_{fa}$ 的关系是什么？

聪明的你也许很快就能想到……

从 $fa$ 到 $i$ ，一共有 $sz_i$ 个点的距离减少了 1，有 $n-sz_i$ 个点的距离加了 1.

所以，可以轻松得到：

$ans_i=ans_{fa}-sz_i+(n-sz_i)$ 。

这样就做完了……

代码

记得开 long long……

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n;
vector<int> g[200001];
int sz[200001],f[200001],sum;
int ans[200001];
void dfs1(int x,int fa)
{
	sz[x]=1;
	for(auto y:g[x])
	{
		if(y!=fa)
		{
			f[y]=f[x]+1;
			dfs1(y,x);
			sz[x]+=sz[y];
		}
	}
}
void dfs2(int x,int fa)
{
	for(auto y:g[x])
	{
		if(y!=fa)
		{
			ans[y]=ans[x]-sz[y]+(n-sz[y]);
			dfs2(y,x);
		}
	}
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	cin>>n;
	int x,y;
	for(int i=1;i<n;i++)
	{
		cin>>x>>y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	
	dfs1(1,1);
	for(int i=1;i<=n;i++)
		sum+=f[i];
	ans[1]=sum;
	dfs2(1,1);
	for(int i=1;i<=n;i++)
		cout<<ans[i]<<" ";
	return 0;
}