两次dfs-大臣的旅费

#include <iostream>
#include <vector>
using namespace std;
int n;
struct road{
  int to,len;
  road(){}
  road(int t,int l){
    to=t,len=l;
  }
};
vector<road> v[100010];
int d[100010];
void dfs(int dis,int id,int fa){
  d[id]=dis;
  for(int i=0;i<v[id].size();i++){
    int u=v[id][i].to;
    if(u!=fa){
      dfs(v[id][i].len+dis,u,id);
    }
  }
}
int main()
{
  cin>>n;
  for(int i=0;i<n-1;i++){
    int x,y,z;
    cin>>x>>y>>z;
    v[x].push_back(road(y,z));
    v[y].push_back(road(x,z));
  }
  dfs(0,1,-1);
  int a=0;
  for(int i=1;i<=n;i++){
    if(d[i]>d[a]){
      a=i;
    }
  }
  dfs(0,a,-1);
  int b=0;
  for(int i=1;i<=n;i++){
    if(d[i]>d[b]){
      b=i;
    }
  }
  long long cost=d[b]*10+((long long)d[b]*(d[b]+1))/2;
  cout<<cost<<endl;
  return 0;
}