假日住宿

题目描述

给出一棵\(N\)节点的树,每个节点代表一个城市,每个城市有一个人,每个人离开自己的城市到另一个城市,每个城市只能有一个人,问这\(N\)个人移动距离和的最大值。

输入格式

输入的第一行包含一个整数
\(T(1<=t<=10)\),表示测试用例的数量。每个测试用例包含几行。 第一行包含一个整数\(2<=N<=10^5\)
,代表城市数。 然后接下来的行分别包含三个整数\(X, Y, Z\)
意味着在城市\(X\)和城市\(Y\)之间有一条高速公路,其长度为\(Z\)。 您可以假设所有城市都已连接并且高速公路是双向的。

输出格式

输出每个样例中,所有人员的最大总行程。

样例

输入样例1
2 
4 
1 2 3 
2 3 2 
4 3 2 
6 
1 2 3
2 3 4
2 4 1
4 5 8
5 6 5
输出样例1
18
62

分析

这道题其实和树的重心有异曲同工的地方,虽然并不是求出重心来进行相关的操作,但是它所用的统计边数的方法与其十分类似。
此题关键不是在于人在怎么走,即不需要考虑人的移动,而是如何把边的作用发挥到最大。
考虑边的情况:
一条边连接了两个顶点,这两个顶点又可以拓展为两棵子树,如果想要最后的路程最长,那么就应该要更多的人去走这一条边,此时最多的人数就是左右两颗子树的点数中的较小者2w(左边的人走到右边,右边的人走到左边),点数就可以用与重心相同的方法计算。

核心代码

void dfs(LL fa, LL u) {
    s[u] = 1;//边界出口 || 初始只有自己一个节点
    for (int i = 0; i < G[u].size(); i++) {
        LL v = G[u][i].v, w = G[u][i].w;
        if (v == fa)
            continue;
        dfs(u, v);
        s[u] += s[v];//累加点数
        ans += w * 2 * min(s[v], n - s[v]);//即左右两颗子树的点数中的较小者 * 2 * w
    }
}