树的直径1

时间限制：1秒        内存限制：128M

题目描述

给定一棵树，树中包含 n个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。
现在请你找到树中的一条最长路径。
要求使用广度优先遍历思想，找到一条路径，使得使得路径两端的点的距离最远。
注意：路径中可以只包含一个点。

输入描述

第一行包含整数 n。
接下来 n−1 行，每行包含三个整数 ai,bi,ci表示点 ai和 bi 之间存在一条权值为 ci 的边。

输出描述

输出一个整数，表示树的最长路径的长度。

样例

输入

6
5 1 6
1 4 5
6 3 9
2 6 8
6 1 7

输出

22

提示

数据范围
1≤n≤10000
1≤ai,bi≤n,
0≤ci≤10^5

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int head[N*2],ver[N*2],Next[N*2],w[N*2],tot=-1,maxx,maxx_root;
void add(int x,int y,int z) {
	ver[++tot]=y;
	w[tot]=z;
	Next[tot]=head[x];
	head[x]=tot;
}
int vis[N];
void dfs(int x,int s) {
	vis[x]=1;
	if(s>maxx) {
		maxx=s;
		maxx_root=x;
	}
	for(int i=head[x]; i!=-1; i=Next[i]) {
		int y=ver[i];
		if(!vis[y]) {
			dfs(y,s+w[i]);
		}

	}
}
int main() {
	memset(head,-1,sizeof(head));
	int n,x,y,z;
	cin>>n;
	for(int i=1; i<n; i++) {
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
	}
	dfs(1,0);
	maxx=0;
	memset(vis,0,sizeof(vis));
	dfs(maxx_root,0);
	cout<<maxx;
	return 0;
}