题目链接:树的最长路径
题目大意:求一棵树中相距最远的两个节点的距离,我们称其为树的直径。
做法:
1、根据图论知识,可以分两步求:
a.找任意点
A
A
A,找到距离A的最远点
U
U
U;
b.再找到距离
U
U
U 最远的
V
V
V,
d
(
U
,
V
)
d(U,V)
d(U,V)就是所求。*
不过最近在练习动态规划,并且从这篇开始双向更新进度,所以从树形动规的角度来分析这题:
2、虽然说是树形动规,但其实本质就是用dfs更新最大值。
即随意指定一个节点为根节点,然后找到以这个节点为根的最大深度和次大深度,相加即可。
不过从中我有点体会是树形动规中比较重要的:
最直白的,状态的转移是一个节点的所有子节点的状态根据规则合法地通过决策转移到该节点上的过程,阶段就是从 $$ i层到
i
−
1
i-1
i−1 层。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=10010;
int e[N*2],ne[N*2],h[N],idx,w[2*N];
int ans;
int n;
void add(int a,int b,int c){
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++; //邻接表存图
}
int dfs(int u,int father){
int dist=0; //dist,d1,d2用途不同,dist相当于在为上一层服务,d1、d2相当于在为本层更新长度
int d1=0,d2=0;
for(int i=h[u];i!=-1;i=ne[i]){
int j=e[i];
if(j==father) continue;
int d=dfs(j,u)+w[i];
dist=max(dist,d);
if(d>d1) d2=d1,d1=d;
else if(d>d2) d2=d;
}
ans=max(ans,d1+d2);
return dist;
}
int main(){
cin>>n;
int a,b,c;
memset(h,-1,sizeof(h));
for(int i=1;i<n;i++){
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
add(b,a,c);
}
dfs(1,-1);
cout<<ans<<endl;
return 0;
}