C语言-算法-最小生成树

讲解了如何使用Kruskal算法解决最小生成树问题,涉及数据结构、图论和算法实现,适用于IT技术相关课程或竞赛
摘要由CSDN通过智能技术生成

【模板】最小生成树

题目描述

如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出 orz

输入格式

第一行包含两个整数 N , M N,M N,M,表示该图共有 N N N 个结点和 M M M 条无向边。

接下来 M M M 行每行包含三个整数 X i , Y i , Z i X_i,Y_i,Z_i Xi,Yi,Zi,表示有一条长度为 Z i Z_i Zi 的无向边连接结点 X i , Y i X_i,Y_i Xi,Yi

输出格式

如果该图连通,则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz

样例 #1

样例输入 #1

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

样例输出 #1

7

提示

数据规模:

对于 20 % 20\% 20% 的数据, N ≤ 5 N\le 5 N5 M ≤ 20 M\le 20 M20

对于 40 % 40\% 40% 的数据, N ≤ 50 N\le 50 N50 M ≤ 2500 M\le 2500 M2500

对于 70 % 70\% 70% 的数据, N ≤ 500 N\le 500 N500 M ≤ 1 0 4 M\le 10^4 M104

对于 100 % 100\% 100% 的数据: 1 ≤ N ≤ 5000 1\le N\le 5000 1N5000 1 ≤ M ≤ 2 × 1 0 5 1\le M\le 2\times 10^5 1M2×105 1 ≤ Z i ≤ 1 0 4 1\le Z_i \le 10^4 1Zi104

样例解释:

所以最小生成树的总边权为 2 + 2 + 3 = 7 2+2+3=7 2+2+3=7

代码

#include <stdio.h>
#include <stdlib.h>
#define MAXN 300000
#define MAXM 300000
typedef struct // 定义边的结构体
{
	int u, v, w;
}Edge;
Edge edges[MAXM]; // 存储所有的边
int pa[MAXN]; // 并查集数组,用于判断两个节点是否在同一棵树中
int N, M; // N是节点数,M是边数
int cmp(Edge a, Edge b); // 边的比较函数,按照边的权重从小到大排序
int find(int x); // 并查集的查找函数,用于查找节点x的根节点
int kruskal(); // Kruskal算法函数

int main(int argc, char *argv[])
{
	int 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);
	}
	int res = kruskal(); // 调用Kruskal算法计算最小生成树的权重
	if (res == -1) // 如果图不连通
	{
		printf("orz\n");
	}
	else
	{
		printf("%d\n", res);
	}
	return 0;
}

int cmp(Edge a, Edge b) // 边的比较函数,按照边的权重从小到大排序
{
	return (a.w - b.w);
}

int find(int x) // 并查集的查找函数,用于查找节点x的根节点
{
	while (pa[x] != x)
	{
		x = pa[x];
	}
	return x; 
}

int kruskal() // Kruskal算法函数
{
	int i, j;
	for (i = 0; i < M; i++) // 将所有的边按照从小到大排序
	{
		for (j = i + 1; j < M; j++)
		{
			if (cmp(edges[i], edges[j]) > 0)
			{
				Edge temp = edges[i];
				edges[i] = edges[j];
				edges[j] = temp;
			}
		}
	}
	for (i = 1; i <= N; i++) // 初始化并查集
	{
		pa[i] = i;
	}
	int res = 0; // 存储最小生成树的权重
	int cnt = 0; // 存储当前已经选择的边的数量
	for (i = 0; i < M; i++) // 遍历所有的边
	{
		int u = find(edges[i].u); // 边的一个节点
		int v = find(edges[i].v); // 边的另一个节点
		// 如果两个节点不在同一棵树中,说明这条边可以添加到最小生成树中
		if (u != v)
		{
			res += edges[i].w; // 更新最小生成树的权重
			pa[u] = v; // 合并两棵树
			cnt++; // 更新已经选择的边的数量
		}
	}
	if (cnt != N - 1) // 如果选择边的数量小于N - 1,说明图不连通
	{
		return -1;
	}
	return res; // 返回最小生成树的权重
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值