题干:
给定一棵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;
}