搜索与图论——Kruskal算法求最小生成树

kruskal算法相比prim算法思路简单,不用处理边界问题,不用堆优化,所以一般稀疏图都用Kruskal。

Kruskal算法时间复杂度O(mlogm)

每条边存结构体里,排序需要在结构体里重载小于号

判断a,b点是否连通以及将点假如集合中需要并查集的知识

cpp 复制代码
#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 100010, M = 200010;

int n, m;
int p[N];

struct Edge
{
    int a, b, w;
    bool operator< (const Edge& W)const
    {
        return w < W.w;
    }
}edges[M];

int find(int x)
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int Kruskal()
{
    int res = 0,cnt = 0;
    for(int i = 1; i <= n; i ++ )
    {
        p[i] = i;
    }
    for(int i = 0; i < m; i ++ )
    {
        int a = find(edges[i].a), b = find(edges[i].b);
        if(a != b)
        {
            p[a] = b;
            res += edges[i].w;
            cnt ++ ;
        }
    }
    if(cnt < n - 1) return 0;
    else return res;
}

int main()
{
    cin >> n >> m;
    for(int i = 0; i < m; i ++ )
    {
        int a, b, w;
        cin >> a >> b >> w;
        edges[i].a = a, edges[i].b = b, edges[i].w = w;
    }
    sort(edges, edges + m);
    int t = Kruskal();
    if(!t) cout << "impossible" << endl;
    else cout << t << endl;
    return 0;
}
相关推荐
wanzhong23335 分钟前
CUDA学习5-矩阵乘法(共享内存版)
深度学习·学习·算法·cuda·高性能计算
fufu031130 分钟前
Linux环境下的C语言编程(四十八)
数据结构·算法·排序算法
Yingye Zhu(HPXXZYY)1 小时前
Solution to Luogu P6340
算法
小熳芋1 小时前
单词搜索- python-dfs&剪枝
算法·深度优先·剪枝
Xの哲學1 小时前
Linux SLAB分配器深度解剖
linux·服务器·网络·算法·边缘计算
bu_shuo1 小时前
MATLAB中的转置操作及其必要性
开发语言·算法·matlab
高洁012 小时前
图神经网络初探(2)
人工智能·深度学习·算法·机器学习·transformer
爱装代码的小瓶子2 小时前
算法【c++】二叉树搜索树转换成排序双向链表
c++·算法·链表
思成Codes2 小时前
数据结构:基础线段树——线段树系列(提供模板)
数据结构·算法
虾..4 小时前
Linux 简单日志程序
linux·运维·算法