kruskal——宿命的PSS

最新推荐文章于 2025-09-06 22:33:46 发布

转载最新推荐文章于 2025-09-06 22:33:46 发布 · 65 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/largecube233/p/6797827.html

文章标签：

#数据结构与算法

本文提供了一道名为Vijos OJ 1579的问题解决方案，采用C++实现并详细展示了通过并查集算法解决此问题的过程。文章中的代码实现了对输入数据进行排序，并利用并查集来处理边权，最终计算出最小生成树的权重。

https://vijos.org/p/1579
题解在这里
http://blog.youkuaiyun.com/fop_zz/article/details/70670069
但是代码我的好

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#define Ll long long
using namespace std;
struct cs{Ll x,y,v;}a[20005];
Ll fa[20005],size[20005];
Ll X,Y,n,ans;
bool cmp(cs a,cs b){return a.v<b.v;}
int get(int x){
    if(!fa[x])return x;
    return fa[x]=get(fa[x]);
}
int main()
{
    scanf("%lld",&n);
    for(int i=1;i<n;i++)scanf("%lld%lld%lld",&a[i].x,&a[i].y,&a[i].v);
    sort(a+1,a+n,cmp);
    for(int i=1;i<=n;i++)size[i]=1;
    for(int i=1;i<n;i++){
        X=get(a[i].x); Y=get(a[i].y);
        ans+=(size[X]*size[Y]-1)*(a[i].v+1)+a[i].v;
        fa[X]=Y; size[Y]+=size[X];
    }
    printf("%lld",ans);
}