1207. 大臣的旅费 dfs 树的直径

现在我们来模拟一下样例：

解析：

1. 首先，我们读取节点数量n=5。
2. 然后，我们读取每条边的信息，并将边添加到对应节点的边列表中。得到的边信息如下：
   • 城市1和城市2之间有一条双向高速路，长度为2千米。
   • 城市1和城市3之间有一条双向高速路，长度为1千米。
   • 城市2和城市4之间有一条双向高速路，长度为5千米。
   • 城市2和城市5之间有一条双向高速路，长度为4千米。
3. 接下来，我们从根节点（城市1）开始进行深度优先搜索，得到每个节点到根节点的距离如下：
   • 城市1的距离为0。
   • 城市2的距离为2。
   • 城市3的距离为1。
   • 城市4的距离为7。
   • 城市5的距离为6。
4. 然后，我们找到距离最远的节点u（城市4），并再次从该节点开始进行深度优先搜索，得到每个节点到城市4的距离如下：
   • 城市4的距离为0。
   • 城市2的距离为5。
   • 城市5的距离为9。
5. 最后，我们再次找到距离最远的节点u（城市5），计算其距离s=9，并根据公式s * 10 + s * (s + 1ll) / 2计算结果，得到135。

所以，最终输出的结果为135。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 100010;

int n;
struct Edge
{
    int id, w;
};
vector<Edge> h[N];
int dist[N];

void dfs(int u, int father, int distance)
{
    dist[u] = distance;

    for (auto node : h[u])
        if (node.id != father)
            dfs(node.id, u, distance + node.w);
}

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        //双向的
        h[a].push_back({b, c});//a到b的距离为c
        h[b].push_back({a, c});//b到a的距离为c
    }

    dfs(1, -1, 0);

    int u = 1;
    for (int i = 1; i <= n; i ++ )
        if (dist[i] > dist[u])
            u = i;

    dfs(u, -1, 0);

    for (int i = 1; i <= n; i ++ )
        if (dist[i] > dist[u])
            u = i;

    int s = dist[u];

    printf("%lld\n", s * 10 + s * (s + 1ll) / 2);

    return 0;
}

为什么要这样做：因为每个城市只能经过一次，且从首都到达每个大城市的方案都是唯一的。

从首都开始遍历，找到最远能到的位置，此点到首都的距离就出来了，