模板 C/C++ 最小生成树 kruskal算法

最新推荐文章于 2024-07-31 11:54:02 发布

原创最新推荐文章于 2024-07-31 11:54:02 发布 · 243 阅读

0 ·

CC 4.0 BY-SA版权

模版同时被 2 个专栏收录

8 篇文章

订阅专栏

图论

4 篇文章

订阅专栏

博客主要围绕某技术展开，包含基本思想和代码实现两部分内容，聚焦信息技术领域相关知识。

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

一、基本思想
二、代码实现

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

int read(){
    int ans = 0, f = 1;
    char c = getchar();
    while(c<'0'||c>'9'){
        if(c=='-') {
            f=-1;
            c=getchar();
        }
    }
    while(c >= '0' && c <='9'){
        ans = ans*10+c-'0';
        c = getchar();
    }
    return f*ans;
}

struct Edge{
    int u;
    int v;
    int w;
}edge[200005];

int fa[5005],n,m,ans,eu,ev,cnt;

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

//并查集循环实现模板，及路径压缩
int find(int x){
    while(x!=fa[x]) x=fa[x]=fa[fa[x]];
    return x;
}

void kruskal(){
    //将边的权值排序
    sort(edge,edge+m,cmp);
    for(int i=0;i<m;i++){
        eu=find(edge[i].u), ev=find(edge[i].v);
        //若出现两个点已经联通了，则说明这一条边不需要了
        if(eu==ev){
            continue;
        }
        cnt ++;
        //将此边权计入答案
        ans+=edge[i].w;
        //将eu、ev合并
        fa[ev]=eu;
        //循环结束条件，及边数为点数减一时
        if(cnt==n-1){
            break;
        }

    }
}
int main(){
    n=read(),m=read();
    //初始化并查集
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
    for(int i=0;i<m;i++){
        edge[i].u=read(),edge[i].v=read(),edge[i].w=read();
    }
    kruskal();
    printf("%d",ans);
    return 0;
}