Prim算法+Kruskal算法笔记

最新推荐文章于 2024-07-19 19:11:19 发布

Enjoycola_

最新推荐文章于 2024-07-19 19:11:19 发布

阅读量240

点赞数 4

分类专栏：算法基础笔记文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/Enjoy10ve/article/details/129131555

版权

算法基础笔记专栏收录该内容

9 篇文章 1 订阅

订阅专栏

一、Prim算法

prim算法理解：

理解之后，我们发现其实prim算法与朴素版的dijkstra算法有很多相同之处

代码模板:

int prim()
{
	memset(dis, 0x3f, sizeof dis);//初始化起点到各点的值为正无穷

	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 || dis[t] > dis[j]))//如果j不在集合内且要么是第一步，要么是有更小的边
				t = j;//更新
		}
		if (i && dis[t] == INF)return INF;//如果当前点的边权值为正无穷，则接着往下还是正无穷
		if (i)res += dis[t];
		st[t] = true;//标记已经过
		for (int j = 1; j <= n; j++)dis[t] = min(dis[j], g[t][j]);

		
	}
	return res;
}

相关例题：

题目来源于Acwing

→原题链接

858. Prim算法求最小生成树

给定一个 n 个点 m 条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

给定一张边带权的无向图 G=(V,E)，其中 V 表示图中点的集合，E表示图中边的集合，n=|V|，m=|E|

由 V 中的全部 n 个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G的一棵生成树，其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。

输入格式

第一行包含两个整数 n 和 m。

接下来 m 行，每行包含三个整数 u,v,w表示点 u 和点 v 之间存在一条权值为 w 的边。

输出格式

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

数据范围

1≤n≤500
1≤m≤105
图中涉及边的边权的绝对值均不超过 1000010000。

输入样例：

输出样例：

Ac代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N =510;
int dis[N],g[N][N];
bool st[N];
int n,m;
int INF =1e9;
int prim()
{
    memset(dis,0x3f,sizeof dis);
    int res=0;
    for(int i =0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!st[j]&&(t==-1||dis[t]>dis[j]))
            t=j;
        }
        if(i&&dis[t]==INF)return INF;
        if(i)res+=dis[t];
        st[t]=true;
        for(int j=1;j<=n;j++) dis[j]=min(dis[j],g[t][j]);
        
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if(i==j)g[i][j]=0;
    else g[i][j]=INF;
    while(m--)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        g[a][b]=g[b][a]=min(g[a][b],w);
        
    }
    int t=prim();//用t临时存放，防止两次调用函数结果冲突
    if(t==INF)printf("impossible");
    else printf("%d",t);
}

二、Kruskal算法

算法思路:

①将每条边的权重按从小到大排序

②枚举每条边a,b权重c,如果a,b不连通，将这条边加到集合中；

代码模板：

int n,m;//n是点数,m是边数
int p[N];//P[N]存储的是父节点;

struct ss     // 存储边
{
    int a, b, w;

    bool operator< (const Edge &W)const
    {
        return w < W.w;
    }
}stu[M];

其中bool operator<(const node&p) const{}

是一个重载小于号运算符的函数，用来比较两个node型变量的大小

int main()
{
	scanf_s("%d%d", &n, &m);
	for (int i = 0; i < m; i++)
	{
		int a, b, w;
		scanf_s("%d%d%d", &a, &b, &w);
		stu[i] = { a,b,w };
	}
	//1.排序
	sort(stu, stu + m);
	//2.更新并查集
	for (int i = 1; i <= n; i++) p[i] = i;
	//3.找连通块
	int res = 0, cnt = 0;
	for (int i = 0; i < m; i++)
	{
		int a = stu[i].a, b = stu[i].b, w = stu[i].w;
		a = find(a), b = find(b);
		if (a != b)//如果a,b不在同一个集合内
		{
			//合并
			p[a] = b;
			//加权值
			res += w;
			cnt++;
		}
	}

相关例题：

例题来自于Acwing

→原题链接

Kruskal算法求最小生成树

给定一个 n 个点 m条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

给定一张边带权的无向图 G=(V,E)，其中 V 表示图中点的集合，E 表示图中边的集合，n=|V|，m=|E|。

由 V 中的全部 n个顶点和 E 中 n−1 条边构成的无向连通子图被称为 G 的一棵生成树，其中边的权值之和最小的生成树被称为无向图 G 的最小生成树。

输入格式

第一行包含两个整数 n 和 m。

接下来 m 行，每行包含三个整数 u,v,w，表示点 u 和点 v 之间存在一条权值为 w 的边。

输出格式

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

数据范围

1≤n≤105
1≤m≤2∗105
图中涉及边的边权的绝对值均不超过 10001000。

输入样例：

输出样例：

Ac代码:

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 200010;
int n, m;
int p[N];
struct ss
{
	int a, b, w;
	bool operator < (const ss &W)const
	{
		return w < W.w;
	}
}stu[N];
int find(int x)
{
	if (p[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);
		stu[i] = { a,b,w };
	}
	//1.排序
	sort(stu, stu + m);
	//2.更新并查集
	for (int i = 1; i <= n; i++) p[i] = i;
	//3.找连通块
	int res = 0, cnt = 0;
	for (int i = 0; i < m; i++)
	{
		int a = stu[i].a, b = stu[i].b, w = stu[i].w;
		a = find(a), b = find(b);
		if (a != b)
		{
			//合并
			p[a] = b;
			//加权值
			res += w;
			cnt++;
		}
	}
	if (cnt < n - 1)printf("impossible");
	else printf("%d\n", res);
}