最小生成树模板(prim+kruskal+prim的优化)

本文详细介绍了两种经典的最小生成树算法:Prim算法和Kruskal算法。Prim算法通过不断扩张包含最短边的集合来逐步构建最小生成树,而Kruskal算法则是按边的权重从小到大排序后,逐条加入不构成环的边来构造最小生成树。文章通过代码实例展示了两种算法的具体实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最小生成树:解决极小连通子图连接所有点使得花费最小问题

1.prim算法 

思想:随机选择一个点,作为初始集合,并保存所有点到这个集合的最短距离,然后找与这个集合距离最短的点,也将其加入这个集合,由于新加入了一个点,所以要更新所有未加入集合点到这个集合的最短距离,不断重复,直到连通整个图。

1)邻接矩阵建图朴素法(无向图)复杂度o(v^2) 

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 111
#define inf 1<<31-1
using namespace std;
int n,m,ans;
int vis[maxn],mapp[maxn][maxn],dis[maxn];//dis记录未被加入集合的点到集合的最短距离
void init()//初始将所有边权值赋为正无穷大
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) mapp[i][j]=inf;
}
void prim()
{
    fill(vis,vis+maxn,0);
    for(int i=1;i<=n;i++) dis[i]=mapp[1][i];//随便选择一个点,这里选择第一个点,dis记录所有点到这个点的距离
    dis[1]=0;//自己到自己距离为0
    vis[1]=1;//标记为访问过
    for(int i=1;i<n;i++)//第一个点已经加入集合,还有m-1个点
    {
        int Min=inf,k;
        for(int j=1;j<=n;j++) if(!vis[j]&&Min>dis[j]) Min=dis[j],k=j;//选择到集合距离最短的点,更新最短距离并记录该点
        if(Min==inf) {ans=-1;return;}//Min为无穷大表明没有与该图连通的点,也就是说这个图不是完全图
        ans+=Min;
        vis[k]=1;//将与该集合距离最短的点标记为访问过
        for(int j=1;j<=n;j++) if(!vis[j]&&dis[j]>mapp[k][j]) dis[j]=mapp[k][j];//更新所有与该集合的最近距离
    }
 }

2)prim+邻接矩阵建图+堆优化

#include<iostream>
#include<cstdio>
#include<queue>
#define maxn 111
#define inf 1<<29
using namespace std;
int vis[maxn],mapp[maxn][maxn],n,m,dis[maxn];
struct node
{
    int v,len;//v表示当前节点,len表示该点到已经连通集合的最短距离
    friend bool operator <(node a,node b)
    {
        return a.len>b.len;
    }
};
int prim()
{
    node cur;
    int ans = 0;
    priority_queue<node>q;
    fill(dis,dis+maxn,inf);
    fill(vis,vis+maxn,0);
    cur.v=1;
    cur.len=0;
    q.push(cur);
    while(q.size())
    {
        cur=q.top();
        q.pop();
        if(vis[cur.v]) continue;
        vis[cur.v]=1;
        ans+=cur.len;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i]&&dis[i]>mapp[i][cur.v])
            {
                node next;
                next.v=i;
                next.len=mapp[i][cur.v];
                dis[i]=mapp[i][cur.v];
                q.push(next);
            }
        }
    }
    return ans;
}

2.kruskal  复杂度o(eloge)

kruskal的思想就是先记录两点和这两个点的权值,然后对所有边进行排序。这里还用到了了并查集,若这两点不属于同一个集合,说明这两个点之间没有路,因为已经是从小到大排了序,所以直接把他们连通,让他们属于同一个集合,不断判断,直到所有边都进行了判断,然后再根据是否只有一个节点的父亲是他本身,若是,则说明最小生成树已经求出,否则不存在这样的树.

以hdu 1863为例

#include<iostream>
#include<algorithm>
#include<cstdio>
#define maxn 111
using namespace std;
int father[maxn];
struct node
{
    int l,r;
    int value;
};
int Find(int x)//并查集查找是否属于同一个集合
{
   if(father[x]!=x)
      father[x]=Find(father[x]);
   return father[x];
}
int Merge(int x,int y)//若不是同一集合,则加入集合
{
    x=Find(x);
    y=Find(y);
    if(x!=y) father[x]=y;
}
bool same(int x,int y)//判断是否是同一集合
{
    return Find(x)==Find(y) ? 1:0;
}
bool rule(node x,node y)
{
    return x.value<y.value;
}
int main()
{
    int n,m;
    node a[maxn];
    while(cin>>m>>n&&m)
    {
        for(int i=1;i<=n;i++) father[i]=i;//并查集初始化
        for(int i=1;i<=m;i++) cin>>a[i].l>>a[i].r>>a[i].value;
        sort(a+1,a+n+1,rule);//排序
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            if(!same(a[i].l,a[i].r))//若两点之间没有路径,连接这条路,即加入同一个集合
            {
                 Merge(a[i].l,a[i].r);
                 ans+=a[i].value;
            }
        }
        int sum = 0;
        for(int i=1;i<=n;i++) if(father[i]==i) sum++;//看是否所有点都连通,若只有一个点的父亲是它本身,则表明所有点已经连通
        if(sum==1) cout<<ans<<endl;
        else cout<<'?'<<endl;
    }
    return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值