Tree（换根dp）

Tree

题意：

看题面。

思路：

换根dp模板题。选任意一点为根节点，dfs一遍求包含当前节点，子树有多少节点，求根节点的答案。再dfs一遍换根dp，对于每个转移，让子节点为根，那么子节点包含子节点以及子树的节点深度都会减1，剩下的点深度都会加1，按照这个思路dfs一遍转移就行，最后输出答案。

#include<bits/stdc++.h>
const int N=1e6+5;
vector<int> G[N];
int son[N],ans[N],d[N],n;
void addedge(int u,int v)
{
	G[u].push_back(v);
	G[v].push_back(u);
}
void dfs(int cur,int fa)
{
	son[cur]=1;
	for(int i=0;i<G[cur].size();i++)
	{
		if(G[cur][i]==fa) continue;
		d[G[cur][i]]=d[cur]+1;
		dfs(G[cur][i],cur);
		son[cur]+=son[G[cur][i]];
		ans[cur]+=ans[G[cur][i]]+d[G[cur][i]];
	}
}
void kdfs(int cur,int fa)
{
	for(int i=0;i<G[cur].size();i++)
	{
		if(G[cur][i]==fa) continue;
		ans[G[cur][i]]=ans[cur]-son[G[cur][i]]+n-son[G[cur][i]];
		kdfs(G[cur][i],cur);
	}
}
int main()
{
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin>>n;
    int u,v;
    memset(son,0,sizeof(son));
    memset(ans,0,sizeof(ans));
    memset(d,0,sizeof(d));
    for(int i=0;i<n-1;i++)
    {
        cin>>u>>v;
        addedge(u,v);
    }
    dfs(1,0);
    kdfs(1,0);
    cout<<*min_element(ans+1,ans+1+n)<<endl;
	return 0;
}