每周题解：走廊泼水节

题目链接

走廊泼水节

题目描述

给定一棵 $N$ 个节点的树，要求增加若干条边，把这棵树扩充为完全图，并满足图的唯一最小生成树仍然是这棵树。

求增加的边的权值总和最小是多少。

注意： 树中的所有边权均为整数，且新加的所有边权也必须为整数。

输入格式

第一行包含整数 $t$，表示共有 $t$ 组测试数据。

对于每组测试数据，第一行包含整数 $N$。

接下来 $N-1$ 行，每行三个整数 $X,Y,Z$，表示 $X$ 节点与 $Y$ 节点之间存在一条边，长度为 $Z$。

输出格式

每组数据输出一个整数，表示权值总和最小值。

每个结果占一行。

样例 #1

样例输入 #1

2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5

样例输出 #1

4
17 

提示

【数据范围】

$1\le N\le 6000$，
$1\le Z\le 100$

算法思想

根据题目描述，将一棵树扩充为完全图，并满足图的唯一最小生成树仍然是这棵树，求增加的边的权值总和最小是多少。

考虑一种增加边的顺序：按照原树边权从小到大的遍历，对于当前边$a\leftrightarrow b$的边权为$c$，将$a$和$b$所属的两个连通块扩充为一个完全图。如下图所示：

此时，需要在两个连通块之间扩充多少条边才能将其扩充为一个完全图呢？不妨假设两个连通块中分别有$s_a$和$s_b$个节点，那么任意两个节点之间都要有边相连，则需要$s_a\times s_b-1$条边，即减去当前边$a\leftrightarrow b$。

确定了边数后，增加的这些边的权值又该如何计算呢？不妨假设新增加的边权为$w$，那么：

• 如果$w<c$， 假设原树为最终完全图的最小生成树，则增加边的过程是按边权从小到大的顺序扩充的，此时新加入边的与原树构成环，可以证明去掉$c$，保留$w$会得到一个更优的生成树，不满足图的唯一最小生成树仍然是这棵树的要求。
• 如果$w=c$，按照上相同思路，可以证明去掉$c$保留$w$会得到总权值相同的最小生成树, 与题意唯一矛盾。
• 因此$w>c$，则新加入边的最小权重为$w=c+1$。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 6005;
struct E {
    int a, b, c;
    bool operator < (const E &e) const
    {
        return c < e.c;
    }
}e[N];
int n, p[N], s[N];
int find(int x)
{
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}
int main()
{
    int T;
    cin >> T;
    while(T --)
    {
        cin >> n;
        for(int i = 1; i <= n; i ++) { p[i] = i; s[i] = 1; }
        for(int i = 1; i < n; i ++) cin >> e[i].a >> e[i].b >> e[i].c;
        sort(e + 1, e + n);
        long long ans = 0;
        for(int i = 1; i < n; i ++)
        {
            int a = e[i].a, b = e[i].b, c = e[i].c;
            a = find(a), b = find(b);
            if(a != b) {
                p[a] = b;
                ans += (long long)(c + 1) * (s[a] * s[b] - 1);
                s[b] += s[a];
            }
        }
        cout << ans << endl;
    }
    return 0;
}