P8602 [蓝桥杯 2013 省 A] 大臣的旅费 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<iostream>
#include <algorithm>
#include <vector>
using namespace std;
#define int long long
const int N=5e5+100;
int n;
int res=0;
typedef pair<int,int>pii;
vector<pii>g[N];
bool vis[N];
int len=0;
int dfs(int u)
{vis[u]=true;int d1=0;//d1记录的是u向下能走的最大长度int d2=0;//d2记录的是u向下能走的第二的长度for(auto i:g[u]){int v=i.first;int w=i.second;if(vis[v])//防止往回搜索continue;int d=dfs(v)+w;if(d>d1)//更新最大值与第二大值d2=d1,d1=d;else if(d>d2)d2=d;}len=max(len,d1+d2);//保存最大的数值return d1;//返回最长长度
}
signed main()
{ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);cin>>n;for(int i=1;i<=n-1;i++){int p,q,d;cin>>p>>q>>d;g[p].push_back((pii){q,d});g[q].push_back((pii){p,d});}dfs(1);//求树的直径，1是根节点int ans=0;for(int i=1;i<=len;i++)//计算答案ans+=10+i;cout<<ans;return 0;
}