走廊泼水节

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

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

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

输入格式
第一行包含整数t，表示共有t组测试数据。

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

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

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

每个结果占一行。

数据范围
1≤N≤60001≤N≤6000
1≤Z≤1001≤Z≤100

输入样例：
2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5 
输出样例：
4
17 

算法解析
根据上面的信息,我们不难发现这道题目和最小生成树算法联系紧密,那么现在
我们的主要问题就在于如何去构造最小生成树.

我们可以考虑最小生成树算法中的Kruskal算法.

首先将所有的边按照从小到大的顺序排序.
此时我们保证了是最小生成树的完美生成法则.

对于每一条边(x,y,w)(x,y,w)而言,他们之间有某种关系.
假如说xx和yy不在同一个连通块(集合)之中,也就是他们之间没有边相连

那么我们相连之后,现在这两个点,各自所在的连通块(集合),都拥有了一个最短边,
也就是(x,y,w)(x,y,w).

最小生成树是已经确定了,但是对于这原来两个连通块的其他点怎么办?
首先我们设Sx表示为x之前所在的连通块 那么Sy表示为y之前所在的连通块.
首先我们设Sx表示为x之前所在的连通块 那么Sy表示为y之前所在的连通块.

因为我们不能破坏这个最小生成树,所以我们这原来的两个连通块中的点就必须
有如下性质.
假如说点A属于Sx这个集合之中 点B属于Sy这个集合之中.
假如说点A属于Sx这个集合之中 点B属于Sy这个集合之中.

那么点AA与点BB之间的距离,必须要大于之前的ww,否则就会破坏之前的最小生成树
所以说(A,B)之间的距离最小为w+1
所以说(A,B)之间的距离最小为w+1
假如说我们知道
Sx有p个元素,然后Sy有q个元素.
Sx有p个元素,然后Sy有q个元素.

那么将
Sx与Sy连通块的所有点相连.
Sx与Sy连通块的所有点相连.

显然这个两个连通块会增加.
p×q−1条边
p×q−1条边

然后每一条边的最小长度为
w+1
w+1

所以我们会得出
(w+1)×(p∗q−1)为两个连通块成为完全图的最小代价

#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>

#define x first
#define y second

using namespace std;

const int N = 6010;
int p[N],Size[N];
struct Edge
{
    int a, b, w;
    bool operator< (const Edge &t) const
    {
        return w < t.w;
    }
}e[N];
int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    int T;
    cin >> T;
    while(T--){
        int n;
        cin>>n;
        for (int i = 0; i < n-1; i ++ ){
             int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            e[i]={a,b,c};
        }
        for (int i = 1; i <= n; i ++ ) p[i]=i,Size[i]=1; 
         
         sort(e,e+n-1);
         int res=0;
        for (int i = 0; i < n - 1; i ++ ){
            int a = find(e[i].a), b = find(e[i].b), w = e[i].w;
            if(a!=b){
                res+=(Size[a]*Size[b]-1)*(w+1);
                Size[b]+=Size[a];
                p[a]=b;
            }
        }
        cout<<res<<endl;
    }
    return 0;
}