题目描述
给定一棵 N 个节点的树,要求增加若干条边,把这棵树扩充为完全图,并满足图的唯一最小生成树仍然是这棵树。
求增加的边的权值总和最小是多少。
注意: 树中的所有边权均为整数,且新加的所有边权也必须为整数。
输入格式
第一行包含整数 t,表示共有 t 组测试数据。
对于每组测试数据,第一行包含整数 N。
接下来 N−1 行,每行三个整数 X,Y,Z,表示 X 节点与 Y 节点之间存在一条边,长度为 Z。
输出格式
每组数据输出一个整数,表示权值总和最小值。
每个结果占一行。
数据范围
1
≤
N
≤
6000
1≤N≤6000
1≤N≤6000
1
≤
Z
≤
100
1≤Z≤100
1≤Z≤100
输入样例:
2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5
输出样例:
4
17
题解
这道题借助Kruskal算法来解决
我们找到一条边权最小的边,将其确定为最小生成树中的边,那么这条边一定是满足两端分别在被连接的两棵树上的边中,边权最小的一条,假设为w,为了防止这条边被其他边更新掉,同时要使增加的边的权值最小,我们将其他满足分别在被连接的两棵树上的边的权值设为w+1
这个操作可以在我们合并两棵树的同时进行,连边的过程是无关紧要的,因为我们只需要权值总和,所以只需要将权值总和增加即可
我们可以存储每棵树的大小siz,在更新答案时,执行操作:
a
n
s
+
=
(
w
+
1
)
∗
(
s
i
z
[
x
]
∗
s
i
z
[
y
]
−
1
)
ans+=(w+1)*(siz[x]*siz[y]-1)
ans+=(w+1)∗(siz[x]∗siz[y]−1)
减1的原因是,我们要将这条最小生成树中的边除去,不对其进行计算
在合并时额外将树的大小与之更新即可
code
#include<bits/stdc++.h>
using namespace std;
const int N=6010;
int T,n,fa[N],siz[N];
struct node
{
int x,y,z;
}edge[N];
bool cmp(node x,node y)
{
return x.z<y.z;
}
int get(int x)
{
if(x==fa[x]) return x;
else return fa[x]=get(fa[x]);
}
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=1;i<n;i++) scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
sort(edge+1,edge+n,cmp);
for(int i=1;i<=n;i++) fa[i]=i,siz[i]=1;
long long ans=0;
for(int i=1;i<n;i++)
{
int fx=get(edge[i].x);
int fy=get(edge[i].y);
if(fx==fy) continue;
ans+=(long long)(edge[i].z+1)*(siz[fx]*siz[fy]-1);
fa[fx]=fy;
siz[fy]+=siz[fx];
}
printf("%lld\n",ans);
}
return 0;
}