【Ybtoj 第22章例3】最长距离【树形DP】

---

解题思路

从每个点出发到达的最远距离 对于一个点的答案有$2$ 种，向子节点走/向父节点走。。

第一遍$dfs$找向子节点走的最远距离，第二遍$dfs$取 $max(子节点走,向父节点走+父节点向下的最大值)$，因为对于一个节点x最长距离可能是经过它的父节点fa 然后向下。

特判：由于路径不能重复，当$x$为$fa$向子节点走的最远节点时，要用向父节点走+父节点向下的次大值，因为如果用最大值相当于先向上再原路返回。。

注意要每次更新次大值，更新完最大值，次大值就是更新前的最大值

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;

int n,xx,yy,k,maxn[10010],head[20010],cmaxn[10010];//cmaxn为次大值,maxn为最大值


struct c{
	int x,next,w;
}a[20010];

void add(int x,int y,int w)
{
	a[++k].x=y;
	a[k].w=w;
	a[k].next=head[x];
	head[x]=k;
}

void dfs(int x,int fa)
{
	maxn[x]=0,cmaxn[x]=0;
	for(int i=head[x]; i; i=a[i].next)
	{
		int y=a[i].x;
		if(y==fa)continue;
		dfs(y,x);
		if(maxn[y]+a[i].w>maxn[x])
		{
			cmaxn[x]=maxn[x];
			maxn[x]=maxn[y]+a[i].w;
		}
		else if(maxn[y]+a[i].w>cmaxn[x])
			cmaxn[x]=maxn[y]+a[i].w;
	}
}

void dfs2(int x,int fa,int w)
{
	if(maxn[fa]!=maxn[x]+w)
	{
		if(maxn[x]<maxn[fa]+w)
		{
			cmaxn[x]=maxn[x];
			maxn[x]=maxn[fa]+w;
		}
		else if(cmaxn[x]<maxn[fa]+w)
			cmaxn[x]=maxn[fa]+w;
	}
	else
	{
		if(maxn[x]<cmaxn[fa]+w)
		{
			cmaxn[x]=maxn[x];
			maxn[x]=cmaxn[fa]+w;
		}
		else if(cmaxn[x]<cmaxn[fa]+w)
			cmaxn[x]=cmaxn[fa]+w;
	}
	for(int i=head[x];i;i=a[i].next)
	{
		int y=a[i].x;
		if(y==fa)continue;
		dfs2(y,x,a[i].w);
	}
		
}

int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		memset(head,0,sizeof(head));
		k=0;
		for(int i=2; i<=n; i++)
		{
			scanf("%d%d",&xx,&yy);
			add(xx,i,yy);
		}
		dfs(1,0);
		for(int i=head[1]; i; i=a[i].next)
			dfs2(a[i].x,1,a[i].w);
		for(int i=1;i<=n;i++)
			printf("%d\n",maxn[i]);
	}

}
/*
5
1 1
2 1
3 1
1 1
*/