这是我的代码:
#include <bits/stdc++.h>
using namespace std;
struct info{
long long to,w;
};
long long dep[100100];
vector<info> g[100100];
long long p=1,q=1;
void dfs(long long x,long long fa){
for(info e:g[x]){
if(e.to==fa) continue;
dep[e.to]=dep[x]+e.w;
dfs(e.to,x);
}
}
int main(){
long long n;
cin>>n;
for(int i=1;i<n;i++){
long long u,v,c;
cin>>u>>v>>c;
g[u].push_back({v,c});
g[v].push_back({u,c});
}
dfs(1,0);
for(int i=1;i<=n;i++){
if(dep[i]>dep[p]) p=i;
}
memset(dep,0,sizeof(dep));
dfs(p,0);
for(int i=1;i<=n;i++){
if(dep[i]>dep[q]) q=i;
}
cout<<dep[q]<<endl;
return 0;
}
题目描述
小Q最近学习了一些图论知识。根据课本,有如下定义。树:无回路且连通的无向图,每条边都有正整数的权值来表示其长度。如果一棵树有N个节点,可以证明其有且仅有N-1 条边。 路径:一棵树上,任意两个节点之间最多有一条简单路径。我们用 dis(a,b)表示点a和点b的路径上各边长度之和。称dis(a,b)为a、b两个节点间的距离。
直径:一棵树上,最长的路径为树的直径。树的直径可能不是唯一的。
现在小Q想知道,对于给定的一棵树,其直径的长度是多少?(以及有多少条边满足所有的直径都经过该边,不需要计算,供思考 )
输入格式
第一行包含一个整数N,表示节点数。
接下来N-1行,每行三个整数a, b, c ,表示点 a和点b之间有一条长度为c的无向边。
输出格式
一行一个整数,表示直径的长度。
数据范围与提示
对于100%的测试数据:2≤N≤200000,所有点的编号都在1..N的范围内,
边的权值≤10^9。
输入输出样例
样例1
输入样例
6
3 1 1000
1 4 10
4 2 100
4 5 50
4 6 100
输出样例
1110
【样例说明】
直径共有两条,3 到2的路径和3到6的路径。这两条直径都经过边(3, 1)和边(1, 4)。