最小生成树—prim和kruskal算法

最小生成树—prim和kruskal算法

一般Kruskal用的比prim多,因为prim可以解决的,kruskal一定可以解决,反过来就不一定了。
n是点的数量,m是边的数量。
prim时间复杂度:O(n^2) 还有一个像Dijkstra的堆优化版本,但是被Kruskal完爆,所以基本不会用。一般用在稠密图(边的数量接近n^2), 用邻接矩阵来存。
如果是用

kruskal时间复杂度:O(mlogm),存储边的时候不是用的邻接表,而是用三元组的形式存起来,方便进行排序。

prim算法

prim 算法采用的是一种贪心的策略。

每次将离连通部分的最近的点和点对应的边加入的连通部分,连通部分逐渐扩大,最后将整个图连通起来,并且边长之和最小。有点类似与Dijkstra算法。
在这里插入图片描述
代码思想和Dijk极其相似,只有dist的定义变了,变成了到集合的最短距离。

代码

#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

const int N=510,INF=0x3f3f3f3f;

int n,m;
int g[N][N]; // 邻接矩阵
int dist[N];// 存放i顶点到集合的最短距离
bool st[N];// 判断i顶点是否在集合中了

int prim()
{
    memset(dist,0x3f,sizeof dist);
    int res=0;//存放最小生成树的长度
    for(int i=0;i<n;i++)//n次遍历,每次加一个顶点到集合中
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;//如果j不在集合中,并且j是第一个顶点或者j到集合的距离更小
        
        if(i&&dist[t]==INF) return INF;//如果图不连通
        if(i) res+=dist[t];//找到第二个顶点的时候就可以在结果上加上这个边长了

        //以刚加入集合的点t,更新其他点到集合的最短距离。但是有点点有可能存在负的自环,dist[t]有可能一直变小,
        //但是已经无所谓了,反正dist[t]在上一步已经被加进去了
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);

        st[t]=true;
            
    }

    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g);

    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        g[a][b]=g[b][a]=min(g[a][b],c);
    }

    int t=prim();

    if(t==INF) puts("impossible");
    else printf("%d\n",t);
    return 0;
}

证明:

在这里插入图片描述
反证法:如果当前与外界直接相连的权值最小的一条边,不出现在最优解中,那么好啊,你再找一条权值最小的边连那个点呗,然后再把我这个边去掉,你这不还是生成树吗?但是不好意思,我已经是权值最小的边了,嘿嘿。

Kruskal算法

如果说prim是一个大的集合逐渐向外扩展,那么Kruskal就是多线作战,最后才合体。
在这里插入图片描述
代码思想:先存下来所有的边,然后对边权进行排序,优先连小的边,直至联通。判断是否联通和连边的操作利用了并查集。最后使用cnt判断连在一起的点数,如果小于n,则证明没有联通,不存在最小生成树。

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 200010;

int n, m;
int p[N]; //并查集中的parent

struct Edge
{
    int a, b, w;

    bool operator<(const Edge &W) const
    {
        return w < W.w;
    }
} edges[N];
//判断两个元素是否属于同一集合,看他们的根节点是否相同即可
int find(int x)//这个函数可以找到x的根节点
{
    // if(x==p[x]) return x;
    // else
    //     return find(p[x]);
    // 路径压缩。
    if (p[x] != x)//如果x的爸爸不等于x,
        p[x] = find(p[x]);//
    return p[x];//如果自己是自己的爸爸,返回自己
}

int main()
{
    scanf("%d%d", &n, &m);

    for (int i = 0; i < m; i++)
    {
        int a, b, w;
        scanf("%d%d%d", &a, &b, &w);
        edges[i] = {a, b, w};
    }
	
    sort(edges, edges + m);

    for (int i = 1; i <= n; i++)
        p[i] = i;//并查集初始化

    int res = 0, cnt = 0;
    for (int i = 0; i < m; i++)
    {
        int a = edges[i].a, b = edges[i].b, w = edges[i].w;
        a=find(a),b=find(b);//分别找打a,b的根节点
        if(a!=b){//如果a,b不属于同一个集合,则合并
            p[a]=b;//让b变成a的爸爸
            res+=w;
            cnt++;
        }
    }
    if(cnt<n-1) puts("impossible");
    else printf("%d\n",res);

    return 0;
}

证明

在这里插入图片描述
连接两个不相连的集合,一定要找权值最小的啊。

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Prim算法Kruskal算法都是用于求解最小生成树的经典算法Prim算法的基本思想是从一个点开始,每次选择一个与当前生成树距离最近的点加入生成树中,直到所有点都被加入生成树为止。具体实现时,可以使用一个优先队列来维护当前生成树与未加入生成树的点之间的距离,每次选择距离最小的点加入生成树中。 Kruskal算法的基本思想是从边开始,每次选择一条权值最小且不会形成环的边加入生成树中,直到生成树中包含所有点为止。具体实现时,可以使用并查集来判断是否形成环。 下面是Prim算法Kruskal算法的C语言代码实现: Prim算法: ```c #include <stdio.h> #include <stdlib.h> #include <limits.h> #define MAX_VERTICES 1000 int graph[MAX_VERTICES][MAX_VERTICES]; int visited[MAX_VERTICES]; int dist[MAX_VERTICES]; int prim(int n) { int i, j, u, min_dist, min_index, sum = 0; for (i = 0; i < n; i++) { visited[i] = 0; dist[i] = INT_MAX; } dist[0] = 0; for (i = 0; i < n; i++) { min_dist = INT_MAX; for (j = 0; j < n; j++) { if (!visited[j] && dist[j] < min_dist) { min_dist = dist[j]; min_index = j; } } u = min_index; visited[u] = 1; sum += dist[u]; for (j = 0; j < n; j++) { if (!visited[j] && graph[u][j] < dist[j]) { dist[j] = graph[u][j]; } } } return sum; } int main() { int n, m, i, j, u, v, w; scanf("%d%d", &n, &m); for (i = 0; i < n; i++) { for (j = 0; j < n; j++) { graph[i][j] = INT_MAX; } } for (i = 0; i < m; i++) { scanf("%d%d%d", &u, &v, &w); graph[u][v] = graph[v][u] = w; } printf("%d\n", prim(n)); return 0; } ``` Kruskal算法: ```c #include <stdio.h> #include <stdlib.h> #include <limits.h> #define MAX_VERTICES 1000 #define MAX_EDGES 1000000 struct edge { int u, v, w; }; int parent[MAX_VERTICES]; struct edge edges[MAX_EDGES]; int cmp(const void *a, const void *b) { return ((struct edge *)a)->w - ((struct edge *)b)->w; } int find(int x) { if (parent[x] == x) { return x; } return parent[x] = find(parent[x]); } void union_set(int x, int y) { parent[find(x)] = find(y); } int kruskal(int n, int m) { int i, sum = 0; for (i = 0; i < n; i++) { parent[i] = i; } qsort(edges, m, sizeof(struct edge), cmp); for (i = 0; i < m; i++) { if (find(edges[i].u) != find(edges[i].v)) { union_set(edges[i].u, edges[i].v); sum += edges[i].w; } } return sum; } int main() { int n, m, i; scanf("%d%d", &n, &m); for (i = 0; i < m; i++) { scanf("%d%d%d", &edges[i].u, &edges[i].v, &edges[i].w); } printf("%d\n", kruskal(n, m)); return 0; } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值