[Tvvj1391]走廊泼水节(最小生成树)

Posted lsgjcya

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[Tvvj1391]走廊泼水节(最小生成树)相关的知识,希望对你有一定的参考价值。

[Tvvj1391]走廊泼水节

Description

给定一棵N个节点的树,要求增加若干条边,把这棵树扩充为完全图,并满足图的唯一最小生成树仍然是这棵树。求增加的边的权值总和最小是多少。
完全图:完全图是一个简单的无向图,其中每对不同的顶点之间都恰连有一条边相连(来自百度百科)

输入格式

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

输出格式

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

样例输入

2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5

样例输出

4
17

数据范围与约定

  • 每个测试点最多10组测试数据
    50% n<=1500;
    100% n<=6000
    100% z<=100

样例解释

第一组数据,在2和3之间修建一条长度为4的道路,是这棵树变成一个完全图,且原来的树依然是这个图的唯一最小生成树.

模拟最小生成树的过程,将初始边排序,对于初始边的两个端点(x,y),为了保证将这两个集合连接起来的边是((x,y)),那么两个集合其他之间的边权(>=v(x,y)),所以对于两个集合的添加边,最小的边权为(v(x,y)+1),所以统计一下两个集合内有多少个点,算出两个集合的添加边为多少即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int read()
{
    int x=0,w=1;char ch=getchar();
    while(ch>'9'||ch<'0') {if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*w;
}
const int N=6010;
int n,cnt,ans;
int head[N],fa[N],size[N];
struct node{
    int x,y,z;
}f[N];
bool cmp(node p,node q){return p.z<q.z;}
void clear()
{
    ans=0;cnt=0;memset(head,0,sizeof(head));for(int i=1;i<=n;i++) fa[i]=i,size[i]=1;
}
int gfa(int x){if(x==fa[x])return x;return fa[x]=gfa(fa[x]);}
int main()
{
    int t=read();
    while(t--)
    {
        n=read();clear();
        for(int i=1;i<n;i++)
        {
            f[i].x=read();f[i].y=read();f[i].z=read();
        }
        sort(f+1,f+n,cmp);
        for(int i=1;i<n;i++)
        {
            int xx=gfa(f[i].x),yy=gfa(f[i].y);
            if(xx==yy) continue;
            ans+=(f[i].z+1)*(size[xx]*size[yy]-1);
            fa[xx]=yy;size[yy]+=size[xx];
        }
        cout<<ans<<endl;
    }
}

以上是关于[Tvvj1391]走廊泼水节(最小生成树)的主要内容,如果未能解决你的问题,请参考以下文章

tyvj1391走廊泼水节——kruskal

TYVJ1391 走廊泼水节

JoyOI1391 走廊泼水节

CH6201 走廊泼水节最小生成树

Contest Hunter CH6201 走廊泼水节 最小生成树 Kruskal

CH6201走廊泼水节