历届试题 大臣的旅费（树的直径）

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int h[N],e[2*N],ne[2*N],w[2*N],idx;
int n;
int d[N];
void add(int a,int b,int v){
	e[idx]=b,ne[idx]=h[a],w[idx]=v,h[a]=idx++;
}
void dfs(int u,int fa,int dis){
	d[u]=dis;
	for(int i=h[u];i!=-1;i=ne[i]){
		int j=e[i];
		if(j==fa) continue;
		dfs(j,u,dis+w[i]);
	}
}
int main(){
	memset(h,-1,sizeof h);
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int x,y,v;
		scanf("%d%d%d",&x,&y,&v);
		add(x,y,v);
		add(y,x,v);
	}
	dfs(1,-1,0);
	int mx=0;
	int k=0;
	for(int i=1;i<=n;i++){
		if(d[i]>mx){
			k=i;
			mx=d[i];
		}
	}
	dfs(k,-1,0);
	mx=0;
	for(int i=1;i<=n;i++){
		mx=max(mx,d[i]);
	}
	printf("%lld\n",1ll*mx*10+1ll*(mx+1)*mx/2);
	return 0;
}