首页 > 编程语言 >Kruskal算法和Prim算法

Kruskal算法和Prim算法

时间:2022-11-25 13:56:43浏览次数:56  
标签:Prim Kruskal 最小 生成 算法 权值 顶点 dis

这两个算法都是求最小生成树的算法,这里有个模板题
P3366 【模板】最小生成树

Kruskal算法与Prim算法相比比较好理解,基于贪心的思想,每次选取不会产生圈(重边)的权值最小的边,直到所有顶点连通。

Kruskal

具体做法:

  1. 将所有顶点看成一个独立的集合
  2. 将边按权值从小打到进行排序
  3. 利用并查集进行分组,如果一个边使得俩个顶点集合联通,那么就把这俩集合分成一组
  4. 直到所有顶点都属于一个集合

如果读者不会并查集,请先去学习并查集再来学最小生成树

这里使用的使优先队列,使用sort也是可以的,重写一下\(compare\)函数即可

  1. 创建一个小根堆,把所有边放入堆中,那么堆顶便是权值最小的边。
  2. \(ans\)为题目结果,cnt为有多少边被使用,最小生成树的边为\(n-1\),其中\(n\)为顶点个数
  3. 从堆顶获取最小的边,判断这个边所对应的俩顶点\((from,to)\)是否在同一组,不在同一组就把他俩合并到同一组里,然后\(++cnt\)并且更新\(ans\)的值
#define ufr(i, x, y) for (int i = x; i <= y; ++i)
struct vertex {
  int from, to, cost;
  bool operator<(const vertex& v) const { return this->cost > v.cost; }
};
  priority_queue<vertex> que;
  ufr(i, 1, M) que.emplace(vs[i]);
  int ans = 0, cnt = 0;
  while (!que.empty()) {
    vertex ver = que.top();
    que.pop();
    if (!same(ver.from, ver.to)) {
      ans += ver.cost;
      ++cnt;
      unit(ver.from, ver.to);
    }
  }

最后一步判断是否可以生成一个最小生成树,就是判断\(cnt\)与\(n-1\)是否相等

Prim

这里存图使用了链式前向星
prim算法与Dijkstra算法比较相似,步骤:

  1. 创建一个\(dis\)数组,,表示最小生成树的起点\(s\),值代表最小生成树中以该顶点为终点的边的权值,初始值设置成无穷大。
  2. 更新以\(s\)为起点的边,并将以这些边为终点的顶点,然后标记s已经访问过(这里跟Bellman Ford中遍历比较相似),更新其在\(dis\)数组中值。
  3. 然后循环遍历,在\(dis\)数组中找一个没访问过,并且在最小生成树中权值最小的顶点(说人话就是没访问过,并且在\(dis\)数组中值最小的索引值)
  4. 重复上面的步骤,直到所有顶点都访问过或者无顶点可以访问

解释一下各变量的意思:

  1. \(now\)就是本次访问的顶点,根据上面的说明,就是当前在最小生成树中,为访问的顶点,并且以该顶点(\(now\))为终点的边以其他未访问过的顶点为终点的边都小
  1. \(ans\)答案,\(tot\)总共已经访问了几个顶点,顶点最多就只能访问N次,所以while(tot++ < N)
  1. \(visited\)数组代表访问过哪些点,\(dis\)数组代表在最小生成树中以索引为终点的边的最小权值
  1. 每次循环都要选择一个顶点,而\(minn\)就代表以那个顶点为终点的边的最小权值,如果一次遍历\(minn==inf\)说明没找到,那么首先\(tot\)没到\(N\),说明处理的顶点少于\(N\),并且没顶点处理了,说明无法生成最小生成树
#define goto(x, y) x + 1, x + y + 1
#define mm(x, y) memset(x, y, sizeof(x))
#define ufr(i, x, y) for (int i = x; i <= y; ++i)
struct {
  int to, w, next;
} edge[MAX_N << 1];
int head[MAX_N], cnt = 0;

int dis[MAX_N];
bool visited[MAX_N];

  fill(goto(dis, N), inf);
  mm(visited, 0);
  int tot = 0, now;
  ll ans = 0;
  dis[1] = 0;
  while (tot++ < N) {
    int minn = inf;
    ufr(i, 1, N) if (!visited[i] && minn > dis[i]) {
      minn = dis[i];
      now = i;
    }
    if (minn == inf) {
      f.pts("orz").ln();
      return;
    }
    visited[now] = true;
    ans += minn;
    for (int i = head[now]; i; i = edge[i].next) {
      if (!visited[edge[i].to] && dis[edge[i].to] > edge[i].w)
        dis[edge[i].to] = edge[i].w;
    }
  }

小根堆优化Prim

标签:Prim,Kruskal,最小,生成,算法,权值,顶点,dis
From: https://www.cnblogs.com/GuanStudy/p/16924887.html

相关文章

  • PHP基于非递归方式算法实现先序/中序/后序遍历二叉树操作
    /** *PHP基于非递归方式算法实现先序/中序/后序遍历二叉树操作 *     A *    B   C *  D  E F   G......
  • 四种语言刷算法之全排列
    力扣46.全排列1、C/***Returnanarrayofarraysofsize*returnSize.*Thesizesofthearraysarereturnedas*returnColumnSizesarray.*Note:Bothre......
  • 9.排序算法
    1.冒泡排序基本介绍:  冒泡排序的基本思想是:通过对待排序序列从前向后(从下标较小的元素开始),一次比较相邻元素的值,如果发现逆序,使较大的元素逐渐从前向后移动。 因为......
  • 算法基础:子矩阵的和
    算法:子矩阵的和以(x1,y1)为左上角,(x2,y2)为右下角的子矩阵的和为:s[x2,y2]-s[x1-1,y2]-s[x2,y1-1]+s[x1-1,y1-1]#include<bits/stdc++.h>using......
  • 算法基础
    算法:前缀和#include<bits/stdc++.h>usingnamespacestd;constintN=1e5+10;inta[N],s[N];//s[N]前缀和数组s[i]=a[1]+a[2]+a[3]+...+a[i]int......
  • 每日算法之二维数组中的查找
    JZ4二维数组中的查找描述在一个二维数组array中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这......
  • 算法5: LeetCode_单链表_两数相加
    题目:*给你两个非空的链表,表示两个非负的整数。它们每位数字都是按照逆序的方式存储的,并且每个节点只能存储一位数字。*请你将两个数相加,并以相同形式返回一个......
  • 算法基础
    算法——差分https://www.acwing.com/problem/content/description/799/#include<bits/stdc++.h>usingnamespacestd;constintN=100010;inta[N],b[N];intm......
  • DBSCAN聚类算法
    1.基于密度的聚类算法基于密度的聚类算法主要思想是只要邻近区域的密度(对象的个数)超过某个阈值,就把它加入到与之相近的聚类中。基于密度的聚类算法代表有DBSCAN算法......
  • 排序算法
    零、总览(一)术语说明稳定:如果a原本在b前面,而a=b,排序之后a仍然在b的前面;不稳定:如果a原本在b的前面,而a=b,排序之后a可能会出现在b的后面;内排序:所有排序操作都在内存中完成......