时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld

题目描述

对于完全图G,若有且仅有一棵最小生成树为T,则称完全图G是树T扩展出的。
给你一棵树T,找出T能扩展出的边权和最小的完全图G。

输入描述:

第一行N表示树T的点数;
接下来N-1行三个整数S i ,T i,D i
描述一条边(S i ,T i )权值为D i ; 保证输入数据构成一棵树。

输出描述:

输出仅一个数,表示最小的完全图GGG的边权和。

示例1
输入

4  
1 2 1  
1 3 1  
1 4 2  

输出

12

说明

添加D(2,3)=2,D(3,4)=3,D(2,4)=3即可。

备注:

对于20%的数据,N≤10;
对于50%的数据,N≤1000;
对于100%的数据,N≤105 ,1≤D i ≤10 5

解题思路:
1、用Kruskal模拟生成树
2、两个集合合成时,构建局部完全图,即假如一个集合的成员点数为n1,另一个为n2,则需要构建n1*n2条边,由贪心思想可以构建一条最小长度d,其余构建长度都为d+1,所以有sum+=d(n1*n2)-1;

不理解欢迎提问哦

代码:

#include<iostream>
#include<algorithm>
using namespace std;
struct Side
{
    int x,y,d;
}side[100010];//储存点信息
bool cmp(Side a,Side b)
{
    return a.d<b.d;
}
int f[100010];
int find(int n)//并查集实现集合
{
    if(f[n]!=n)return f[n]=find(f[n]);
    return n;
}
int main()
{
    int m,i,j,x,y,z;
    long long ans[100010],sum=0;//ans[]记录集合的点数
    for(i=0;i<100010;i++)
    {
        f[i]=i;
        ans[i]=1;
    }
    cin>>m;
    for(i=0;i<m-1;i++)
        cin>>side[i].x>>side[i].y>>side[i].d;
    sort(side,side+i,cmp);
    for(i=0;i<m-1;i++)//Kruskal算法
    {
        int a=find(side[i].x);
        int b=find(side[i].y);
        if(a!=b)
        {
            f[b]=a;
            sum+=(side[i].d+1)*(ans[a]*ans[b])-1;
            ans[a]+=ans[b];
        }
    }
    cout<<sum<<endl;
}

更多推荐