洛谷 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;
}