【USACO3-1-1】最短网络 kruskal

最新推荐文章于 2021-08-13 16:10:57 发布

原创最新推荐文章于 2021-08-13 16:10:57 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#usaco

USACO 同时被 2 个专栏收录

57 篇文章

订阅专栏

图论-kruskal

4 篇文章

订阅专栏

本文介绍了一种使用Kruskal算法求解最小生成树问题的方法。通过定义边结构体和并查集，实现了对给定节点间的连接进行排序，并利用贪心策略选择权重最小且不形成环的边来构建最小生成树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;

struct edge{int x,y,z;}a[101000];
int fa[101000];
int n,tot=0,ans;

bool cmp(edge p,edge q){return p.z<q.z;}
int getfa(int x){return fa[x]==x?fa[x]:fa[x]=getfa(fa[x]);}
void kruskal()
{
    int cnt=0;
    sort(a,a+tot,cmp);
    for (int i=0;i<tot && cnt<n-1;i++)
    {
        int pp=getfa(a[i].x),qq=getfa(a[i].y);
        if (pp!=qq)
        {
            fa[pp]=qq;
            ans+=a[i].z;
            cnt++;
        }
    }
}

int main()
{
    cin>>n;
    for (int i=1;i<=n;i++) fa[i]=i;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            int cur;
            cin>>cur;
            if (!cur) continue;
            a[tot].x=i;a[tot].y=j;a[tot++].z=cur;
        }
    kruskal();
    cout<<ans<<endl;
    return 0;
}