【图论杂题】：C. tree [最小生成树]

题目：

团队链接：tree

题面：

给定一颗$n$个点的带边权的树$T$，要求构造出完全图$G$满足：

• $G$的边权和尽量小，
• $T$为$G$的唯一最小生成树，

要求输出$G$的边权和。
范围：$n<=10^6$

样例：

输入：

10
5 10 1
5 4 4
4 9 6
2 9 6
8 4 6
1 5 2
3 10 9
5 7 7
6 9 3

输出：

319

题解：

这个题，可以巧妙地运用一下最小生成树Kruscal的用法，考虑一下我们求最小生成树的过程中，每次取出连接不同联通块的最小的一条边，合并改变连接的两个联通块，直到所有点联通，现在是要保证树上的每一条边都是连接不同联通块的最小边，按树边的边权从小到大的顺序合并点集，合并点解的过程中把非树边补上，计算边权和即可 。

代码：

#include<bits/stdc++.h>
#define LL long long
using namespace std;
inline int read()
{
	int s=0,w=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=1e6+7;
struct see{int x,ver,next,edge;}e[sea<<1];
int n,tot,fa[sea],size[sea],last[sea];
LL ans;
bool cmp(see a,see b){return a.edge<b.edge;}
int get(int x){if(fa[x]==x) return x;else return fa[x]=get(fa[x]);}
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		x=read(); y=read(); z=read();
		e[i].x=x;e[i].ver=y;e[i].edge=z;
	}
	sort(e+1,e+n,cmp);
	for(int i=1;i<=n;i++) fa[i]=i,size[i]=1;
	for(int i=1;i<n;i++)
	{
		int x=get(e[i].x),y=get(e[i].ver),z=e[i].edge;
		ans+=z; ans+=((LL)(size[x]*size[y]-1LL)*(z+1LL));
		//（size是子树大小，x子树*y子树-1（1是正在枚举的该点））*（边权+1（既然要构造最小的边权，由于是正整数，直接+1即可））（乘法原理即可）
		fa[y]=x;size[x]+=size[y];
	}
	printf("%lld\n",ans);
	return 0;
}

Continue……