HDU 4679 多校8 D题 树型DP

写得好恶心，比赛的时候别人看着我写还没什么感觉，刚才自己又写了一遍各种细节不清→ →

两个DFS

第一个：

DP[u][0]记录以u为根的子树的直径，DP[u][1]记录以结点u的儿子为根的所有子树的直径最大值，dp[u][2]记录以结点u的儿子为根的所有子树的直径次大值

L[u][0]记录以u为根的子树的所有叶子到u的最远距离，L[u][1]为次远，L[u][2]为第三远，要求三个距离来自三个不同的儿子

第二个：蛋疼转移

细节看代码

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
#pragma comment(linker, "/STACK:32000000")
#define N 100010
int dp[N][3],L[N][3];
int head[N],cnt,ans,id;
struct edge
{
	int v,w,next;
}e[N<<1];
int n;
void add(int u,int v,int w)
{
	e[cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs1(int u,int f)
{
	int v;
	dp[u][0]=dp[u][1]=dp[u][2]=0;
	L[u][0]=L[u][1]=L[u][2]=0;
	for(int i=head[u];i!=-1;i=e[i].next)
	if((v=e[i].v)!=f)
	{
		dfs1(v,u);
		if(L[v][0]+1>L[u][0]) L[u][2]=L[u][1],L[u][1]=L[u][0],L[u][0]=L[v][0]+1;
		else if(L[v][0]+1>L[u][1]) L[u][2]=L[u][1],L[u][1]=L[v][0]+1;
		else if(L[v][0]+1>L[u][2]) L[u][2]=L[v][0]+1;
		dp[u][0]=max(max(dp[u][0],L[u][0]+L[u][1]),dp[v][0]);
		if(dp[v][0]>dp[u][1]) dp[u][2]=dp[u][1],dp[u][1]=dp[v][0];
		else if(dp[v][0]>dp[u][2]) dp[u][2]=dp[v][0];
	}
}
void dfs(int u,int f,int l,int d,int ed)
{
	int v,ans0,l0,d0;
	if(ed!=-1)
	{
		ans0=max(d,dp[u][0]);
		if(ans0*e[ed].w<ans || ans0*e[ed].w==ans && id>(ed>>1)+1)
		{
			ans=ans0*e[ed].w;
			id=(ed>>1)+1;
		}
	}
	for(int i=head[u];i!=-1;i=e[i].next)
	if((v=e[i].v)!=f)
	{
		l0=l+1;
		if(L[u][0]==L[v][0]+1) l0=max(L[u][1]+1,l0);
		else l0=max(L[u][0]+1,l0);
		d0=d;
		if(L[u][0]==L[v][0]+1) d0=max(d0,L[u][1]+l);
		else d0=max(d0,L[u][0]+l);
		if(L[u][0]==L[v][0]+1) d0=max(d0,L[u][1]+L[u][2]);
		else if(L[u][1]==L[v][0]+1) d0=max(d0,L[u][0]+L[u][2]);
		else d0=max(d0,L[u][0]+L[u][1]);
		dfs(v,u,l0,d0,i);
	}
}
int main ()
{
freopen("1.in","r",stdin);
	int tt,_=0;
	scanf("%d",&tt);
	while (tt--)
	{
		scanf("%d",&n);
		cnt=0;
		memset(head,-1,sizeof(head));
		for(int i=1;i<n;++i)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}
		dfs1(1,0);
//for(int i=1;i<=n;++i)printf("%d %d %d  %d %d %d\n",dp[i][0],dp[i][1],dp[i][2],L[i][0],L[i][1],L[i][2]);
		ans=id=2147483647;
		dfs(1,0,0,0,-1);
		printf("Case #%d: %d\n",++_,id);
	}
	return 0;
}