最小生成树 kruskal算法 C++实现

最新推荐文章于 2025-10-27 01:30:12 发布

原创最新推荐文章于 2025-10-27 01:30:12 发布 · 2.4k 阅读

2 ·

CC 4.0 BY-SA版权

图专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用Kruskal算法求解最小生成树问题的方法。通过定义节点结构体并运用排序与并查集，实现了对加权无向图的最小生成树计算。程序首先读取顶点数与边数，接着输入每条边的两个顶点及其权重，并将这些信息存入节点数组。随后，根据边的权重进行排序，最后利用并查集确定最小生成树。

#include <iostream>
#include <vector>
#include <algorithm>
#define Inf 0x7fffffff
using namespace std;
struct node
{
    int u,v;
    int w;
    node(int a,int b,int x):u(a),v(b),w(x){}
};
vector<node> edge;
int vn,an;
int pa[1000];
bool cmp(const node &a,const node &b)
{
    return a.w<b.w;
}
int find(int x)
{
    return x==pa[x]?x:pa[x]=find(pa[x]);
}
int kruskal()
{
    int sumw=0,cnt=0;
    for(int i=1;i<=vn;++i) pa[i]=i;
    for(int i=0;i<an;++i)
    {
        int x=find(edge[i].u),y=find(edge[i].v);
        if(x!=y)
        {
            pa[x]=y;
            sumw+=edge[i].w;
            ++cnt;
        }
        if(cnt==vn-1) break;
    }
    return sumw;
}
int main()
{
    cin>>vn>>an;
    for(int i=0;i<an;++i)
    {
        int a,b,x;
        cin>>a>>b>>x;
        edge.push_back(node(a,b,x));
    }
    sort(edge.begin(),edge.end(),cmp);
    cout<<kruskal();
    return 0;
}