【换根DP例题】——深度和最大 P3478

---

题目

原题链接

描述

给定一个 n 个点的树，请求出一个结点，使得以这个结点为根时，所有结点的深度之和最大。一个结点的深度之定义为该节点到根的简单路径上边的数量。

输入格式

• 第一行有一个整数，表示树的结点个数 n。
• 接下来 (n−1) 行，每行两个整数 u,v，表示存在一条连接u,v 的边。

输出格式

• 输出一行一个整数表示你选择的结点编号。如果有多个结点符合要求，输出节点编号的最小值

输入/输出例子

输入

8
1 4
5 6
4 5
6 7
6 8
2 4
3 4

输出

7

提示

无

---

解题思路

分析

这题如果直接暴力做n遍树形DP，时间肯定炸了，所以我们的目标是：
在根结点切换时，直接通过一些已经计算过的数据在O(1)就能得到另一个根的结果。
如图是一棵树：

上图是以1为根节点时各节点的深度。

再看图：
这是以2为根节点时各节点深度。

对比发现：所有2号点的子树的深度都-1，同时除了2号点的子树外，其他的点的深度都+1

动态转移方程

定义：f[i]为i节点为根时的所有深度和,d[i]表示以i为根节点的所有子节点的个数（包含自身）
则根据上文，我们可以得出：
f[v]=f[u]−(d[v])+(n−d[v])
其中，v为子节点，u为父节点

代码实现

先用一遍dfs预处理d数组，再用一遍dfs处理f数组，最后输出。

code

#include<bits/stdc++.h>
#define mod 1
#define int long long
#define foru(i,a,b) for(int i=a;i<=b;i++)
#define ford(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e6+1;
int idx,head[N],n,m,d[N],f[N],x,y,z;
struct fy
{
	int w,to,next;
}edge[N<<1];
char gc()
{
  	static char now[1<<20],*S,*T;
  	if(T==S)
  	{
		T=(S=now)+fread(now,1,1<<20,stdin);
    	if(T==S) 
			return EOF;
  	}
  return *S++;
}
template <typename T>
void Read(T&x) 
{
  	x=0;
  	char c=gc();
  	while(c<'0'||c>'9') 
  		c=gc();
  	x=c-'0';
  	while((c=gc())>='0'&&c<='9') 
  		x=x*10+c-'0';
}
template <typename T, typename... Args>
void Read(T&x,Args&...args)
{
  	Read(x);
  	Read(args...);
}
inline void add(int u,int v,int w)
{
	edge[idx].w=w,edge[idx].to=v,edge[idx].next=head[u];head[u]=idx++;
}
void dfs(int u,int fa)//预处理d
{
	d[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa)
			continue;
		dfs(v,u);
		d[u]+=d[v];
	}
}
void dfs_(int u,int fa)//处理f
{
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa)
			continue;
		f[v]=f[u]+n-2*d[v];
		dfs_(v,u);
	}
}
signed main()
{
	memset(head,-1,sizeof(head));
	Read(n);
	foru(i,1,n-1)
		Read(x,y),add(x,y,1),add(y,x,1);
	dfs(1,0);
	foru(i,1,n)
		f[1]+=d[i];
	dfs_(1,0);
	int ans=0,sum=0;
	foru(i,1,n)
		if(f[i]>ans)
			ans=f[i],sum=i;
	printf("%lld",sum);
}