题目链接
背景
P.S.S:“我来自哪里?”
WH:“你来自一个图。”
P.S.S:“我是谁?”
WH:“你是最小生成树。”
P.S.S:“我又要到哪里去?”
WH:“你要成为一个最小完全图(边权之和最小的完全图)。”
P.S.S:“为……为什么啊?”
WH:“这是你的宿命!因为你无聊!!!P.S.S!”
描述
最小生成树P.S.S在宿命的指引下找到了巫师Kismi。P.S.S希望Kismi能帮自己变成一个完全图。Kismi由于某些不可告人的原因,把这件事交给了你。
PS: 可以保证,这个最小生成树对于最后求出的完全图是唯一的。
格式
输入格式
输入的第一行是一个整数n,表示生成树的节点数。
接下来有n-1行,每行有三个正整数,依次表示每条边的端点编号和边权。
(顶点的边号在1-n之间,边权< maxint)
输出格式
一个整数ans,表示以该树为最小生成树的最小完全图的边权之和。
样例1
样例输入1
3
1 2 4
2 3 7
Copy
样例输出1
19
Copy
样例2
样例输入2
4
1 2 1
1 3 1
1 4 2
Copy
样例输出2
12
Copy
限制
每个测点1s
提示
n<20000
来源
冰尘e溶化邀请赛第四题,吴豪原创题。
我们可以发现我们在用Kruskal算法时,我们在构造一颗最小生成树时,我们总是用并查集先判断一下两个集合是否在一起(即是不是连在一棵树上了),如果不在一起,然后我们会将它们合起来(设中间连接的边(u,v)的距离是val),则这两个连起来的集合(设为A和B)相互的距离dis[A[i],B[j]]都大于val(即dis[A[i],B[j]]>val(A[i],B[j]为对于集合内任意节点)),如果dis[A[i],B[j]]<=val,那么我们可以连接A[i]和B[i],而不一定是u,v了。
那么我们就可以来做此题了,我们在实现Kruskal时,我们用number数组记录一下并查集中一个集合中节点的个数。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,f[20010];
long long number[20010];
struct node
{
int u,v;
long long z;
}a[20010];
bool cmp(node a,node b)
{
if (a.z<b.z) return true;
return false;
}
int get(int x)
{
if (f[x]==x) return x;else f[x]=get(f[x]);
return f[x];
}
int main()
{
scanf("%d",&n);
long long ans=0;
for (int i=1;i<=n-1;i++)
{
scanf("%d%d%lld",&a[i].u,&a[i].v,&a[i].z);
ans+=a[i].z;
}
sort(a+1,a+n,cmp);
for (int i=1;i<=n;i++) f[i]=i,number[i]=1;
for (int i=1;i<=n-1;i++)
{
if (get(a[i].u)!=get(a[i].v))
{
long long num=number[get(a[i].u)];
ans+=(a[i].z+1)*(number[get(a[i].u)]*number[get(a[i].v)]-1);
f[get(a[i].u)]=get(a[i].v);
number[get(a[i].v)]+=num;
}
}
printf("%lld\n",ans);
return 0;
}