Distance Sums 2

Distance Sums 2
题意：一棵树求所有 i 点到其他点的距离和。
思路：dfs先求出第一个点，再根据父节点和子节点的关系推出字节点的ans。(换根dp？) ans[i]=ans[fa]+n-2*son[i];

#include<bits/stdc++.h>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=2e18+100;
const int maxn=2e5+100;
int ans[maxn],son[maxn];
vector<int>g[maxn];
int n;
void dfs(int x,int fa,int dep)
{
	son[x]=1;
	for(auto it:g[x])
	{
		if(it==fa)continue;
		ans[1]+=dep;
		dfs(it,x,dep+1);
		son[x]+=son[it];
	}
}
void dfs2(int x,int fa)
{
	if(x!=1)ans[x]=ans[fa]+n-2*son[x];
	for(auto it:g[x])
	{
		if(it==fa)continue;
		dfs2(it,x);
	}
}
signed main()
{
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0,1);
	dfs2(1,0);
	for(int i=1;i<=n;i++)
	{
		cout<<ans[i]<<"\n";
	}
}