1. 程式人生 > >[POI2008]STA-Station

[POI2008]STA-Station

很明顯而簡單的樹形$dp$:

設以節點$v$為根時的深度和

我們可以先處理出以$1$為根的深度和$f[1]$,那麼我們怎麼樣才能不$dfs n$次來求出以其他點為根的深度和呢

我們考慮現在節點為$u$,子節點為$v$,那麼當$v$為根的時候,$f[v]=f[u]+n-2*siz[v]$($size[v]$為$v$子樹大小),為什麼呢,畫張圖,我們把$v$當做根,就是在把$u$當做根的基礎上,$v$的子樹深度都減少了$1$,總共減少了$siz[v]$,而其他節點的深度都$+1$,總共加了$n-siz[v]$

程式碼:

#include<iostream>
#include<cstdio>
#define N 1000007
#define ll long long
using namespace std;
struct Edge
{
	int to,nxt;
}edge[N<<1];
int n,cnt;
int head[N],siz[N],dep[N];
ll f[N];
void Add(int u,int v)
{
	edge[++cnt]=(Edge){v,head[u]};
	head[u]=cnt;
}
void Dfs(int u,int fa)
{
	siz[u]=1;
	dep[u]=dep[fa]+1;
	f[1]+=dep[u];
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].to;
		if(v==fa)
			continue;
		Dfs(v,u);
		siz[u]+=siz[v];
	}
}
void Dp(int u,int fa)
{
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].to;
		if(v==fa)
			continue;
		f[v]=f[u]+n-2*siz[v];
		Dp(v,u);
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;++i)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		Add(u,v);
		Add(v,u);
	}
	Dfs(1,0);
	Dp(1,0);
	ll ans=0,maxn=0;
	for(int i=1;i<=n;++i)
		if(maxn<f[i])
			ans=i,maxn=f[i];
	printf("%lld",ans);
	return 0;
}