#2334 树的直径（树形DP）

原博主博客（还有广搜的写法）

描述

树的直径，即这棵树中距离最远的两个结点的距离。每两个相邻的结点的距离为1，即父亲结点与儿子结点或儿子结点与父子结点之间的距离为1.有趣的是，从树的任意一个结点a出发，走到距离最远的结点b，再从结点b出发，能够走的最远距离，就是树的直径。树中相邻两个结点的距离为1。你的任务是：给定一棵树，求这棵树中距离最远的两个结点的距离。

输入

输入共n行 第一行是一个正整数n，表示这棵树的结点数 接下来的n-1行，每行三个正整数a,b,w。表示结点a和结点b之间有一条边，长度为w 数据保证一定是一棵树，不必判错。

输出

输出共一行 第一行仅一个数，表示这棵树的最远距离

样例输入

4
1 2 10
1 3 12
1 4 15

样例输出

27

提示

• 数据范围和注释 Hint
• 10%的数据满足1<=n<=5
• 40%的数据满足1<=n<=100
• 100%的数据满足1<=n<=10000 1<=a,b<=n 1<=w<=10000

解析

看一眼，是最长路，差点没甩一个SPFA走人
在网上看了一些题解，想法总结一下：

• 和之前做的星座那道题有点像（好像那一道还难些）
• 根据题干，最长路不一定过根节点，但好像也可以暴力枚举每个点作为起点过（没试过，纯口糊^ _ ^）
• 深搜一遍所有的点，从叶子节点开始，更新至父节点，维护f[i][k]（i节点的 第k长的路径长度）只需维护最长和次长

CODE

#include<bits/stdc++.h>。
using namespace std;
const int N=1e4+5;
int read(){
	int s=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-f;
		ch=getchar();
	}
	while(isdigit(ch)){
		s=(s<<3)+(s<<1)+ch-'0';
		ch=getchar();
	}
	return s*f;
}
struct fjy{
	int nxt,v,w;
}e[N<<1];
int fir[N],f[N][3];
int ans,n,u,v,cnt,w;
void add(int ui,int vi,int wi){
	e[++cnt].nxt=fir[ui];
	e[cnt].v=vi;
	e[cnt].w=wi;
	fir[ui]=cnt;
}
void dp(int u,int fa){
	f[u][1]=f[u][2]=0;
	for(int i=fir[u];i;i=e[i].nxt){
		
		if(e[i].v==fa) continue;
		dp(e[i].v,u);//printf("u=%d;",u);
		if(f[u][1]<f[e[i].v][1]+e[i].w){
			f[u][2]=f[u][1];
			f[u][1]=f[e[i].v][1]+e[i].w;//printf("->1:f1=%d,v=%d;",f[u][1],e[i].v);
			//if(f[u][2]<f[e[i].v][2]+e[i].w&&f[u][1]>f[e[i].v][2]+e[i].w) f[u][2]=f[e[i].v][2]+e[i].w;由于最长路与次长路不能有公共边，则若取了v为最长路上的一点，便不能取其为此段路上的点//,printf("->1->1:f2=%d,v=%d;\n",f[u][2]);
			//else printf("->1->2:f2=%d,v=%d;\n",f[u][2]);
		}
		else if(f[u][2]<f[e[i].v][1]+e[i].w) f[u][2]=f[e[i].v][1]+e[i].w;//,printf("->2->1:f2=%d,v=%d;\n",f[u][2]);else printf("->2->2:f2=%d,v=%d;\n",f[u][2]);
	}
	if(ans<f[u][1]+f[u][2]) ans=f[u][1]+f[u][2];
}
int main(){
	n=read();
	for(int i=1;i<n;i++){
		u=read(),v=read(),w=read();
		add(u,v,w),add(v,u,w);//printf("%d\n",e[5].w);
	}
	//printf("w=%d\n",e[fir[1]].w);
	dp(1,0);
	printf("%d",ans);
	return 0;
}