【ybt高效进阶3-2-4】构造完全图

构造完全图

题目链接：ybt高效进阶3-2-4

题目大意

对于一个完全图，它有一个唯一的最小生成树，现在给出这个最小生成树。
问你原来的完全图的所有权值和最小可以是多少。

思路

看到题目中有最小生成树，自然会想用最小生成树的方法来做。

那我们考虑怎么弄。
首先，我们看到这个最小生成树是唯一的，那也就是说只有这个树是最小生成树。其它都构成不了。

那我们想想构造最小生成树的过程：把边从小到大排序枚举，然后如果不在同一个连通块内就连接。
而且它构造的是树，也就是说一旦它用 $i-1$ 条边把 $i$ 个点连了起来，这 $i$ 个点之间就不能再连别的边。

又因为这个最小生成树是唯一的，那我们可以得知说这 $i$ 个点之间的其它边的长度一定是大于连接这 $i$ 个点的 $i-1$ 条边的任何一条。
（因为如果小于，那还不如选这条，如果是等于，就可以选这个也可以选那个，就不是唯一的了）

那我们就把已知的最小生成树上的边按权值从小到大排序枚举，每次这条边会连接两个连通块，假设两个连通块的大小分别是 $x,y$，边的权值是 $z$，那别的边的权值就一定要大于 $z$。而你又要完全图边权值最小，那就选 $z+1$ 的长度。

那我们看看这样可以多连多少条边。首先，因为是完全图，大小为 $x$ 的连通块的每个点都跟大小为 $y$ 的连通块的每个点都连有一条边，那就是 $x\times y$ 条。
但是有一条是树上的，它的长度就不是 $z+1$，而是 $z$。那我们不妨在读入树的时候就加上这个值，然后在这里直接不算。

那新加上的值就是 $(x\times y-1)\times (z+1)$。

求连通块的大小很容易弄，把并查集改成带权并查集，多记录一个连通块大小即可。

记得加上原来树上每条边的权值。

代码

#include<cstdio>
#include<algorithm>

using namespace std;

struct node {
	int x, y;
	long long z;
}a[100001];
int n, fa[100001], num[100001];
long long ans;

bool cmp(node x, node y) {
	return x.z < y.z;
}

int find(int now) {
	if (fa[now] == now) return now;
	return fa[now] = find(fa[now]);
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i < n; i++) {
		scanf("%d %d %lld", &a[i].x, &a[i].y, &a[i].z);
		ans += a[i].z;
	}
	
	sort(a + 1, a + (n - 1) + 1, cmp);
	
	for (int i = 1; i <= n; i++) {
		fa[i] = i;
		num[i] = 1;
	}
	
	for (int i = 1; i < n; i++) {
		int X = find(a[i].x), Y = find(a[i].y);
		
		ans += 1ll * (num[Y] * num[X] - 1ll) * (a[i].z + 1ll);
		//能新增加的边数乘上这些边最小要的权值
		
		fa[X] = Y;//最小生成树操作
		num[Y] += num[X];//统计一个连通块中有多少个点
	}
	
	printf("%lld", ans);
	
	return 0;
}