城邦【第十二届蓝桥杯省赛第二场C/C++】【最小生成树】【Kruskal】【Prim】

Derrickhang

已于 2022-04-02 23:09:46 修改

阅读量332

点赞数

文章标签： c++ 数据结构算法

于 2022-04-02 22:32:12 首次发布

本文链接：https://blog.csdn.net/derrickhang/article/details/123930515

版权

本篇博客详细介绍了第十二届蓝桥杯省赛第二场C/C++中关于最小生成树的问题。通过Kruskal算法，实现了城邦问题的解题思路和AC代码，重点讲解了如何计算边权和使用并查集避免环的产生。此外，还提供了其他解题方法，如Prim算法，供读者参考。

摘要由CSDN通过智能技术生成

一、题目链接

城邦【第十二届蓝桥杯省赛第二场C/C++】【最小生成树】
在这里插入图片描述

二、题目分析

（一）算法标签

最小生成树 Kruskal

（二）解题思路

三、AC代码

解法一:

#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 2021;

LL ans = 0;
int cnt = 0;

struct Edge {
	int a, b, w;
	bool operator< (const Edge& e) const
	{
		return w < e.w;
	}
}edges[N * N / 2];

// 计算边权 
int cost(int a, int b)
{
	int res = 0;
	while (a || b)
	{
		int x = a % 10;
		int y = b % 10;
		if (x != y) res += x + y;
		a /= 10;
		b /= 10;
	}
	return res;
}

int p[N];
int find(int x)
{
	if (p[x] != x) p[x] = find(p[x]);
	return p[x];
}

// 将a,b所在的集合合并 
void merge(int a, int b)
{
	a = find(a), b = find(b);
	p[a] = b;
}

LL kruskal()
{
	// 初始化并查集 
	for (int i = 1; i <= N; i ++ )
		p[i] = i;
	// 按边权从小到大排序 
	sort(edges, edges + cnt);
	// 遍历每条边，如果不连通（不在一个集合），则合并a,b所在的两个集合 
	LL res = 0;
	int num = 0;
	for (int i = 0; i < cnt; i ++ )
	{
		int a = edges[i].a, b = edges[i].b, w = edges[i].w;
		if (find(a) != find(b))
		{
			merge(a, b);
			num ++ ;
			res += w;
		}
	}
	if (num < N - 1) return 0x3f3f3f3f;
	return res;
}

int main()
{
	for (int i = 1; i <= N; i ++ )
		for (int j = i + 1; j <= N; j ++ )
			edges[cnt ++ ] = {i, j, cost(i, j)};	// 两个点之间存一条边即可
	
	ans = kruskal();
    printf("%lld", ans);	// 4046
    return 0;
}