【题解】 HLOJ 1253 走廊泼水节

题目传送门

题目描述：
我们一共有N个OIER打算参加这个泼水节，同时很凑巧的是正好有N个水龙头(至于为什么，我不解释)。N个水龙头之间正好有N-1条小道，并且每个水龙头都可以经过小道到达其他水龙头(这是一棵树，你应该懂的…)。
但是OIER们为了迎接中中的挑战，决定修建一些道路(至于怎么修，秘密),使得每个水龙头到每个水龙头之间都有一条直接的道路连接(也就是构成一个完全图呗)。
但是OIER们很懒，并且记性也不好，他们只会去走那N-1条小道，并且希望所有水龙头之间修建的道路，都要大于两个水龙头之前连接的所有小道(小道当然要是最短的了)。所以神COW们，帮那些OIER们计算一下吧，修建的那些道路总长度最短是多少，毕竟修建道路是要破费的

---

Input Format
本题为多组数据
第一行t，表示有t组测试数据
对于每组数据
第一行N，表示水龙头的个数（当然也是OIER的个数）；
2到N行，每行三个整数X,Y,Z，表示水龙头X和水龙头Y有一条长度为Z的小道。

---

Output Format
对于每组数据，输出一个整数，表示修建的所有道路总长度的最短值。

---

分析：
题意可以转化为给你一个唯一的最小生成树，让你添加边，使整张图变成一张完全图，所添加的最小边权和是多少。

那么我们思考：

• 这个最小生成树是唯一的最小生成树，说明最小权值的完全图也是唯一的。我们想Kruscal的思想:将图按照边权从小到大排序，之后每次筛选最小边，如果不在同一个集合，就加上这条边。

这题已经给了我们唯一的最小生成树，我们是否可以逆推Kruscal的方法呢？

用$num[i]$表示祖先为i的集合的点的个数
用$fa[i]$表示第i个点祖先

先将边按照从小到大的顺序排序，枚举每条边，如果两条边不在同一个集合内，那么我们就可以用开始加边。

设这条边的起点的祖先是$x$ , 终点的祖先是$y$，边权是$v$
对于$x$和$y$两个集合，其中的每条边都可以向另一个集合的每条边连一个权值为$v+1$的边（因为这是唯一的最小生成树，所以如果连v的话就违背了$唯一$这个条件）。

即$ans+=v\ast (num[x]+num[y]-1)$
(-1是因为两个集合中已经有原先的一条边相连，那么这条边不能算。)

做n-1遍后每个点都在同一个集合中，此时这张图也是一张完全图,$ans$是我们需要的答案

那么具体代码如下：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
struct node{
    int x,y;
	LL v;
}e[200001];
int num[100001];
int fa[100001];
int t,n;
LL ans=0;
int getfa(int k){
    return fa[k]==k?k:fa[k]=getfa(fa[k]);
}
inline bool mycmp(node x,node y){
    return x.v<y.v;
}
int main(){
	freopen("input.in","r",stdin);
	freopen("input.out","w",stdout);
    scanf("%d",&t);
    while (t--){
		scanf("%d",&n);
	    for (int i=1;i<=n;i++) fa[i]=i,num[i]=1;
	    ans=0;
	    for (int i=1,x,y,z;i<n;i++)
	      scanf("%d %d %lld",&e[i].x,&e[i].y,&e[i].v),ans+=e[i].v;
	    sort(e+1,e+n,mycmp); 
		for (int i=1;i<n;i++){
		    int x=getfa(e[i].x),y=getfa(e[i].y);
		    if (x==y) continue;
		    fa[x]=y;
		    ans+=(LL)(e[i].v+1)*(num[x]*num[y]-1);
		    num[y]+=num[x];
		}
		printf("%lld\n",ans);
	}
	fclose(stdin);
	fclose(stdout);
	return 0;
}