【树形dp】P3478 [POI2008]STA-Station

这道题目要求换根，先跑一次dfs，求出dep和size

观察到如果将root从u换成u的一个儿子v，那么v的子树（包括v）的dep将会-1，一共-size[v]，其余节点的dep都+1，共计n-size[v]

这样就可以再跑一次dfs，求出最大的位置即可

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int n,cnt,head[maxn];
struct edge
{
	int to,nxt;
}G[maxn<<1];
long long f[maxn],dep[maxn],size[maxn];
void add(int x,int y)
{
	G[++cnt].nxt=head[x]; G[cnt].to=y; head[x]=cnt;
}
void dfs(int u ,int fa)
{
	size[u]=1;
	dep[u]=dep[fa]+1;
	for(int i=head[u];i;i=G[i].nxt)
	{
		if(G[i].to!=fa)
		{
			dfs(G[i].to,u);
			size[u]+=size[G[i].to];
		}
	}
}
void ddfs(int u,int fa)
{
	for(int i=head[u];i;i=G[i].nxt)
	{
		if(G[i].to!=fa)
		{
			f[G[i].to]=f[u]-(size[G[i].to]<<1)+n;
			ddfs(G[i].to,u);
		}
	}
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d",&n);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dfs(1,1);
	for(int i=1;i<=n;i++)
		f[1]+=dep[i];
	ddfs(1,1);
	long long ans=0,id;
	for(int i=1;i<=n;i++)
		if(ans<f[i])
			ans=f[i],id=i;
	printf("%d",id);
	return 0;
}