2.21数据与结构算法学习日记(最小生成树prim算法)

目录

最小生成树prim

最小生成树算法是一种用来在一个加权连通图中找到最小生成树的算法。最小生成树是一个包含图中所有顶点的树,其总权值最小。

prim算法

洛谷题目示例

[P3366 【模板】最小生成树](#P3366 【模板】最小生成树)

题目描述

输入格式

输出格式

输入输出样例

说明/提示

题目分析

代码示例


最小生成树prim

最小生成树算法是一种用来在一个加权连通图中找到最小生成树的算法。最小生成树是一个包含图中所有顶点的树,其总权值最小。

常见的最小生成树算法包括:

  1. Prim算法:从一个顶点开始,每次选择与当前生成树最近的顶点加入生成树中,直到所有顶点都被加入。Prim算法的时间复杂度为O(V^2)或O(ElogV),其中V为顶点数,E为边数。

  2. Kruskal算法:将所有边按权值从小到大排序,依次加入生成树中,但要保证加入的边不会形成环。Kruskal算法的时间复杂度为O(ElogE)或O(ElogV),其中V为顶点数,E为边数。

这两种算法都可以用来求解最小生成树,选择哪种算法取决于具体情况和图的规模。( 这里主要讲解第一种算法)

prim算法

Prim算法是一种用来在加权连通图中找到最小生成树的贪心算法。算法的基本思想是从一个初始顶点开始,逐步扩展生成树,每次选择与当前生成树最近的顶点加入生成树中,直到所有顶点都被加入。

具体步骤如下:

  1. 选择一个初始顶点作为生成树的根节点,并将其加入生成树中。

  2. 从生成树中的所有顶点出发,找到与生成树中的顶点相连且权值最小的边,将其连接的顶点加入生成树。

  3. 重复上述步骤,直到所有顶点都被加入生成树为止。

Prim算法的时间复杂度取决于如何实现最小堆数据结构,通常为O(V^2)或O(ElogV),其中V为顶点数,E为边数。Prim算法适用于稠密图或边的数量接近顶点数的情况。

总的来说,Prim算法是一种简单且有效的算法,用来求解加权连通图的最小生成树问题。

洛谷题目示例

P3366 【模板】最小生成树

题目描述

如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出 orz

输入格式

第一行包含两个整数 N,M,表示该图共有 N 个结点和 M 条无向边。

接下来 M 行每行包含三个整数 Xi​,Yi​,Zi​,表示有一条长度为 Zi​ 的无向边连接结点 Xi​,Yi​。

输出格式

如果该图连通,则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz

输入输出样例

输入 #1复制

复制代码
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出 #1复制

复制代码
7

说明/提示

数据规模:

对于 20%的数据,N≤5,M≤20。

对于 40% 的数据,N≤50,M≤2500。

对于 70%的数据,N≤500,M≤104。

对于 100%100% 的数据:1≤N≤5000,1≤M≤2×105,1≤Zi​≤104。

样例解释:

所以最小生成树的总边权为 2+2+3=7

题目分析

1,模板题,不过注意测试数据存在平行边,如果用邻接矩阵存图的话应该存最短的平行边。

代码示例

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define  Maxr 5001
int n,m;
int mp[Maxr][Maxr];
bool vis[Maxr];
int dist[Maxr];
int Prim()
{
    int sum=0;
    memset(dist,0x7f,sizeof(dist));
    vis[1]=true;
    for(int i=1; i<=n; i++)dist[i]=mp[1][i];//首先把第一个顶点录入,所以下一个循环是i<n
    for(int i=1; i<n; i++)//由于第一个顶点已经录入权值,所以循环n-1次
    {
        int minn=0x7f7f7f7f,minj=0;
        for(int j=1; j<=n; j++)
        {
            if(!vis[j]&&dist[j]<minn)//未走过的节点中权最小的节点的位置
            {
                minn=dist[j];
                minj=j;
            }
        }
        if(!minj)return -1;
        sum+=minn;//累加提前,避免闭环
        vis[minj]=true;
        for(int j=1; j<=n; j++)
        {     //依次更新后面没走过的点的权最小值
            if(!vis[j])dist[j]=min(dist[j],mp[minj][j]);
        }

    }

return sum;
}
int main()
{
    memset(mp,0x7f,sizeof(mp));
    cin>>n>>m;
    for(int i=0; i<m; i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        mp[x][y]=mp[y][x]=min(mp[x][y],z);//邻接矩阵存图
    }

    int sum=Prim();
    if(sum==-1)cout<<"orz"<<endl;
    else cout<<sum<<endl;
    return 0;





}

如果以上有啥问题,还望大佬能指正下哈

相关推荐
是小Y啦15 分钟前
leetcode 106.从中序与后续遍历序列构造二叉树
数据结构·算法·leetcode
@月落19 分钟前
alibaba获得店铺的所有商品 API接口
java·大数据·数据库·人工智能·学习
liuyang-neu25 分钟前
力扣 42.接雨水
java·算法·leetcode
6230_29 分钟前
git使用“保姆级”教程1——简介及配置项设置
前端·git·学习·html·web3·学习方法·改行学it
y_dd32 分钟前
【machine learning-12-多元线性回归】
算法·机器学习·线性回归
m0_6312704032 分钟前
标准c语言(一)
c语言·开发语言·算法
万河归海42832 分钟前
C语言——二分法搜索数组中特定元素并返回下标
c语言·开发语言·数据结构·经验分享·笔记·算法·visualstudio
小周的C语言学习笔记37 分钟前
鹏哥C语言36-37---循环/分支语句练习(折半查找算法)
c语言·算法·visual studio
y_dd37 分钟前
【machine learning-七-线性回归之成本函数】
算法·回归·线性回归
小魏冬琅1 小时前
K-means 算法的介绍与应用
算法·机器学习·kmeans