《算法竞赛进阶指南》0x62 T1 走廊泼水节

题目传送门

题目描述

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

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

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

输入格式

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

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

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

输出格式

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

每个结果占一行。

数据范围

$1\le N\le 6000$
$1\le Z\le 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，在更新答案时，执行操作：
$ans+=(w+1)\ast (siz[x]\ast 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;
}