AcWing 346. 走廊泼水节(最小生成树)

题干：

给定一棵N个节点的树，要求增加若干条边，把这棵树扩充为完全图，并满足图的唯一最小生成树仍然是这棵树。
求增加的边的权值总和最小是多少。
输入格式
第一行包含整数t，表示共有t组测试数据。
对于每组测试数据，第一行包含整数N。
接下来N-1行，每行三个整数X,Y,Z，表示X节点与Y节点之间存在一条边，长度为Z。
输出格式
每组数据输出一个整数，表示权值总和最小值。
每个结果占一行。
数据范围
N≤6000,Z≤100

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

思路：

完全图：即图中任意两点都有直接相连的路径。
因为原有的最小生成树不能变，所以我们新加的边权需要大于已有的。
考虑普通的Kruskal算法，先将边权升序排列，然后合并并查集，其中加一点操作。

若存在集合a<-b<-c<-x，和集合d<-e<-f<-y（即x的父节点为a，y的父节点为d）
当遍历到x-y这条边（权值为z）时，我们需要使两个合并，则需要使所有的点相连，一共需要连（4*4-1）条边且这些边需要比x-y这条边的权值大，因为要求新加边权值和的最小值且z是两个集合中最大的权值，所以权值应为z+1。
反证法：若新加的权值now<=z，则会存在例如x-<d这条边同样可以使两个集合连通且原最小生成树发生改变，则与题意矛盾。

这样我们需要额外记录的就是每个集合中点的个数（即有多少点以当前点为父节点）。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
struct stu{
	int x,y,z;
}edge[6010];
int sum[6010],fa[6010];
int find(int now){
	if(now==fa[now])	return now;
	else	return fa[now]=find(fa[now]);
}
bool cmp(stu a,stu b){
	return a.z<b.z;
}
int main()
{
	ll ans;
    int n,t1;
    scanf("%d",&t1);
    while(t1--){
    	ans=0;
    	scanf("%d",&n);
    	for(int i=0;i<n-1;i++){
    		scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
    	}
    	for(int i=0;i<=n;i++){
    		fa[i]=i;
    		sum[i]=1;
    	}
    	sort(edge,edge+n-1,cmp);
    	for(int i=0;i<n-1;i++){
    		int x=find(edge[i].x);
    		int y=find(edge[i].y);
    		if(x==y) continue;
    		ans+=(long long)(edge[i].z+1)*(sum[x]*sum[y]-1);
    		fa[y]=x;
    		sum[x]+=sum[y];
    	}
    	printf("%lld\n",ans);
    }
    return 0;
}