洛谷 P3478 [POI2008]STA-Station(bzoj P1131 [POI2008]STA-Station)

傳送門

這題必須要吐槽！！！

首先樹形dp先求以1爲根的深度和，然後由父親節點轉移到兒子節點，也就是除了兒子節點以下包括兒子節點的其他節點深度減1，兒子節點以下包括兒子節點的其他節點深度加一（畫個圖很容易看出，實在不行看popoqqq的博客，解釋的很清楚），然後o（1）轉移。狀態轉移方程：f[x]=f[fa[x]]+n-2*son[x]，son[x]就是x節點包括自己的所有兒子節點個數，fa[x]就是x節點的父親。

然後就是吐槽！！！：首先要卡long long，wa兩發，然後注意存圖數組開兩倍，因爲樹無向，re兩發，最後最令人氣憤的是有多個節點最大要輸出編號最小的那個！！神tm英文題面沒看就看了看中文題面然後他又不說。。然後無限wa！！！垃圾題目！！

代碼：

#include<iostream>
#include<stdio.h>
using namespace std;
const int Maxn=1000005;
struct node
{
	int v;
	int next;
}t[2*Maxn];
int head[Maxn*2],deep[Maxn*2],son[Maxn*2];
long long maxn,f[Maxn*2];
int cnt,n,maxi=1;
int Insert(int u,int v)
{
	cnt++;
	t[cnt].v=v;
	t[cnt].next=head[u];
	head[u]=cnt;
}
int dfs_deep(int u,int d)
{
	f[1]+=d;
	deep[u]=d;
	for(int i=head[u];i;i=t[i].next)
		if(deep[t[i].v]==0&&t[i].v!=1)
			son[u]+=dfs_deep(t[i].v,d+1);
	return son[u];
}
void dfs_f(int u)
{
	for(int i=head[u];i;i=t[i].next)
		if(f[t[i].v]==0)
		{
			f[t[i].v]=f[u]+(n-son[t[i].v])-son[t[i].v];
			dfs_f(t[i].v);
		}
} 
int main()
{
	int u,v;
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++)
	{
		scanf("%d%d",&u,&v);
		Insert(u,v);
		Insert(v,u);
	}
	for(int i=1;i<=n;i++)
		son[i]=1;
	dfs_deep(1,0);
	maxn=f[1];
	dfs_f(1);
	for(int i=1;i<=n;i++)
		if(maxn<f[i])
			{
				maxi=i;
				maxn=f[i];
			}
	printf("%d",maxi);
	return 0;
}