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;





}

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

相关推荐
虾球xz7 分钟前
游戏引擎学习第55天
学习·游戏引擎
古希腊掌管学习的神9 分钟前
[搜广推]王树森推荐系统笔记——曝光过滤 & Bloom Filter
算法·推荐算法
qystca10 分钟前
洛谷 P1706 全排列问题 C语言
算法
浊酒南街15 分钟前
决策树(理论知识1)
算法·决策树·机器学习
就爱学编程23 分钟前
重生之我在异世界学编程之C语言小项目:通讯录
c语言·开发语言·数据结构·算法
oneouto23 分钟前
selenium学习笔记(二)
笔记·学习·selenium
学术头条28 分钟前
清华、智谱团队:探索 RLHF 的 scaling laws
人工智能·深度学习·算法·机器学习·语言模型·计算语言学
sealaugh3228 分钟前
aws(学习笔记第十九课) 使用ECS和Fargate进行容器开发
笔记·学习·aws
炭烤玛卡巴卡1 小时前
学习postman工具使用
学习·测试工具·postman
Schwertlilien1 小时前
图像处理-Ch4-频率域处理
算法