最小生成树 (Prime算法+Kruskal算法)

Prime算法:

关于算法的过程演示:

Prime算法的原理是贪心,通过对图上的点进行操作来产生最小生成树。

这个过程可以用两个集合的变化来表现出来,假设有一个集合U用来存放已加入最小生成树的点,另一个集合V存放未加入最小生成树的点。
因为要构造一棵最小生成树,只要两点之间可以相互到达就好,那么每次我们从V中取出来的点只要满足到U中某个点的距离比V中其他点到U中所有点的距离都要小,那么这个点就应该放进U中。为什么呢?因为每个点都要放进去才能构造出最小生成树,那么我们每次放进去的点与U中可以与它相连的点应该是最小的。

P3366 【模板】最小生成树(洛谷)
具体代码实现如下:

#include <iostream>
#include <algorithm>
#include <math.h>
#include <cstdio>
#include <cmath>
#include <stdlib.h>
#include <string.h>
using namespace std;

int n,m,from,to,w;
int G[5005][5005],dis[5005]; //G表示用邻接矩阵来存图,dis[i]表示集合中与i直接相连的点到i
bool vis[5005];             //的最小距离,vis[i]表示该点是否已在最小生成树集合中;
int Max=0x3f3f3f3f;
int Prime()
{
    memset(dis,0x3f,sizeof(dis));//把当前不能到达的点初始化为无穷大;
    dis[1]=0,vis[1]=1;        //把1看成起点并收入最小生成树集合中(可随意找一个点作为起点);
    int t=1,sum=0;           
    for(int i=1; i<=n-1; i++)
    {
        for(int j=1; j<=n; j++)
            dis[j]=min(dis[j],G[t][j]); //更新最短路径中的点可以到达的点的距离;
        t=-1;                        
        for(int j=1; j<=n; j++) //找未加入的点中到已加入点距离最小的;      
        {
            if(vis[j]==0&&(t==-1||dis[t]>dis[j]))
                t=j;
        }
        if(dis[t]==Max) //找不到就直接返回; 
            return Max;
        vis[t]=1;   //把找到的点加入最短路径中;  
        sum+=dis[t];
    }
    return sum;
}
int main()
{
    cin>>n>>m;
    memset(G,0x3f,sizeof(G)); 
    for(int i=1; i<=m; i++)
    {
        cin>>from>>to>>w;
        G[from][to]=G[to][from]=min(G[from][to],w); //存图;
    }
    int ans=Prime();
    if(ans==Max)
        cout<<"orz"<<endl;
    else
        cout<<ans<<endl;
    return 0;
}

Kruskal算法:

添加链接描述

Kruskal算法通过处理边来得到最小生成树,其算法思想是贪心。
记录下图中的每一条边,然后每次取权值最小的边,同时把这条边的两个端点加入到最小生成树,使它们有同一个根节点,每次加入一条边之前要检查两个端点是否有相同的根节点,如果有相同的根节点是不能把这条边加入的,因为这回导致回路的产生,从而无法构成最小生成树。

P3366 【模板】最小生成树(洛谷)
具体代码实现如下:

#include <iostream>
#include <algorithm>
#include <math.h>
#include <cstdio>
#include <cmath>
#include <stdlib.h>
#include <string.h>
using namespace std;

int n,m,id,pre[5005]; //pre[i]表示i点的父节点;
struct node
{
    int from,to,w;
} e[200005];    //用来保存每条边的信息;

bool cmp(node a,node b) //按权值给边排序;
{
    return a.w<b.w;
}
int Find(int x)  //找x的根节点(并查集);
{
    if(x==pre[x])
        return x;
    else
        return pre[x]=Find(pre[x]);
}
void initial()   //pre初始化;
{
    for(int i=1; i<=n; i++)
        pre[i]=i;
}
void Kruskal()
{
    initial();    //初始化父节点集合;
    sort(e,e+id,cmp);   //按权值由小到大排序;
    int cnt=0,sum=0;    //cnt记录最小生成树的边数,sum记录总权值;
    for(int i=0; i<id; i++)
    {
        int u=e[i].from,v=e[i].to,w=e[i].w; //取边并查找端点的根节点;
        int fu=Find(u),fv=Find(v);
        if(fu==fv)
            continue;
        else
        {
            cnt++;
            sum+=w;
            pre[fu]=fv;  //加入并查集;
        }
    }
    if(cnt==n-1)
        printf("%d\n",sum);
    else
        printf("orz\n");
}
int main()
{
    cin>>n>>m;
    for(int i=1; i<=m; i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        e[id++]= {a,b,c};  //存边;
    }
    Kruskal();
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值