最小生成树(prim算法和Kruskal算法)(C++)

题目1(prim算法)

收获

1:学到了关于memset(vis,0,sizeof(vis));初始化函数的使用,只有设置成0,才是都初始化为0,其他的如果写10,并不会像我们理想状态都设置为10 的(参考链接),所以这里想要用最大距离作为判断的时候,直接写INT_MAX,会都赋值为-1,这里采用,都设置成0x3f的方法,最终和它define的INF进行比较,若相等则代表并不能生成最小生成树~
2:关于vectora作为函数参数的调用方法自己还是不是很会,只能设置成全局变量,希望在未来的学习中可以知道是为什么?

代码

在这里插入图片描述

输入:
6 9
0 1 34
0 2 46
0 5 19
1 4 12
5 4 26
2 5 25
2 3 17
3 5 25
3 4 38
运行截图:
在这里插入图片描述

如果对应的没有最小生成树的话,就会输出最大值
在这里插入图片描述

在这里插入图片描述

#include <iostream>
#include <bits/stdc++.h>

using namespace std;

typedef pair<int,int> pii;
vector<pii>a[100];
#define INF 0x3f3f3f3f

int prim(int n,int dist[],int vis[]){
    dist[0]=0;
    long long sum=0;
    for(int i=0;i<n;i++){
        int cur=-1;
        for(int j=0;j<n;j++)
           if(!vis[j]&&(cur == -1||dist[j]<dist[cur]))
              cur=j;
        if(dist[cur]>=INF)
            return INF;
        cout<<dist[cur]<<endl;
        sum+=dist[cur];
        vis[cur]=1;
        for(auto it=a[cur].begin();it!=a[cur].end();it++){
            if(!vis[it->first])
                {dist[it->first]=min(dist[it->first],it->second);
                }
        }
    }
    return sum;
}

int main()
{
    int n,m;
    cin>>n>>m;
    int dist[n];
    int vis[n];
    memset(vis,0,sizeof(vis));
    memset(dist,0x3f,sizeof(dist));
    while(m--){
        int x,y,w;
        cin>>x>>y>>w;
        a[x].push_back(make_pair(y,w));
        a[y].push_back(make_pair(x,w));
    }
    cout << prim(n,dist,vis)<< endl;
    return 0;
}


题目2(Kruskal算法)

收获

代码

参考链接:https://blog.csdn.net/qq_43619271/article/details/109091314

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值