[最小生成树][并查集]JZOJ 2940 生成输入数据

Posted mastervan

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[最小生成树][并查集]JZOJ 2940 生成输入数据相关的知识,希望对你有一定的参考价值。

 

Description

首先看到题目别太开心,这题可不是让你出数据~^_*

背景神马的就忽略了。这题就是给你一棵带边权的树,然后这棵树是某个完全图唯一的最小生成树。问原来的完全图中所有边可能的最小边权和是多少。

完全图是任意两个点之间都有边相连的图。


 

 

Input

第一行包含一个整数T表示数据组数。

每组数据第一行一个整数N表示点数。

接下来N-1行每行三个整数ai,bi,wi表示最小生成树上ai和bi之间有一条权值为wi的边。


 

Output

输出应有T行,每行表示一组数据的答案。


 

 

Sample Input

2
3
1 2 4
2 3 7
4
1 2 1
1 3 1
1 4 2

Sample Output

19
12
 

Data Constraint

 
 

Hint

20%的数据满足:T≤5,n≤5,wi≤5

另外30%的数据满足:n≤1000,给定的树是一条链

100%的数据满足:T≤10,n≤20000,wi≤10000


By moreD

 

分析

题目和题面无关系列

考场的时候不知道为什么总觉得可以有奇怪的方法,打了一波计算用set维护然后20分WA

事实上只用模仿生成最小生成树的过程即可

我们合并两棵树时,我们可以确定其中必定只有一条边是最小生成树内的,需要添加的边数=两棵树大小的乘积-1,边的权值为最小生成树树边的权值+1

然后再加上原树总值就算出来了

 

技术分享图片
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=2e4+10;
struct Edge {
    int u,v,w;
}g[N];
int f[N],sz[N];
int T,n;

int Get_F(int x) {return f[x]==x?x:f[x]=Get_F(f[x]);}

bool CMP(Edge a,Edge b) {
    return a.w<b.w;
}

int main() {
    for (scanf("%d",&T);T;T--) {
        scanf("%d",&n);
        for (int i=1;i<n;i++) scanf("%d%d%d",&g[i].u,&g[i].v,&g[i].w),sz[i]=1,f[i]=i;
        f[n]=n;sz[n]=1;long long ans=0;
        sort(g+1,g+n,CMP);
        for (int i=1;i<n;i++) {
            int fu=Get_F(g[i].u),fv=Get_F(g[i].v);
            if (fu!=fv) {
                ans+=1ll*(sz[fu]*sz[fv]-1ll)*(g[i].w+1ll);
                sz[fu]+=sz[fv];
                f[fv]=fu;
                ans+=g[i].w;
            }
        }
        printf("%lld
",ans);
    }
}
View Code

 

以上是关于[最小生成树][并查集]JZOJ 2940 生成输入数据的主要内容,如果未能解决你的问题,请参考以下文章

最小生成树个数 并查集压缩路径

数据结构与算法—并查集Kruskal算法求最小生成树

最小生成树问题(并查集解决)

1863 畅通工程-并查集最小生成树

Aizu-2224Save your cats并查集+最小生成树

CSP 201703-4 地铁修建 python 最小生成树,并查集