传送门
描述
给定一棵树,树中包含 n 个结点(编号1~n)和 n−1 条无向边,每条边都有一个权值。
现在请你找到树中的一条最长路径。
换句话说,要找到一条路径,使得使得路径两端的点的距离最远。
注意:路径中可以只包含一个点。
输入格式
第一行包含整数 n。
接下来 n−1 行,每行包含三个整数 ai,bi,ci,表示点 ai 和 bi 之间存在一条权值为 ci 的边。
输出格式
输出一个整数,表示树的最长路径的长度。
数据范围
1≤n≤10000,
1≤ai,bi≤n,
−105≤ci≤105
输入样例:
6
5 1 6
1 4 5
6 3 9
2 6 8
6 1 7
输出样例:
22
这道题我们首先考虑建边的时候建无向边,然后我们根节点任取一个都行,然后对于每一个节点找到他的子节点分别的最大长度,然后取两个最
大的,对于每一个节点我们的答案都要进行更新,我们设m1是当前节点子节点的最大值,m2是当前节点的次大值,然后就有
ans=max(ans,m1+m2),这样就能保证我们找到的答案是最大值,然后返回给上一层的最大值就是m1
注意,m1是返回给上层的最大值,而ans是我们答案的最大值,因为不一定点多距离就大
#####AC代码如下:
#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int h[maxn],e[maxn],ne[maxn],w[maxn];
int n,m,cnt,ans=0;
bool vis[maxn];
void add(int x,int y,int p) {
w[cnt]=p,e[cnt]=x,ne[cnt]=h[y],h[y]=cnt++;
}
int dfs(int x) {
int res=0,m1=0,m2=0;
vis[x]=true;
for(int i=h[x]; i!=-1; i=ne[i]) {
int y=e[i];
if(!vis[y]) {
int s=dfs(y);
s+=w[i];
if(s>m1)swap(m1,s);
if(s>m2)swap(m2,s);
}
}
ans=max(m1+m2,ans);
return m1;
}
int main() {
cin>>n;
for(int i=0; i<maxn; i++)h[i]=-1;
for(int i=1; i<n; i++) {
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
dfs(1);
cout<<ans<<endl;
return 0;
}