问题描述
小蓝国是一个水上王国,有2021个城邦,依次编号1到2021,在任意两个城邦之间,都有一座桥直接连接。
为了庆祝小蓝国的传统节日,小蓝国政府准备将一部分桥装饰起来。
对于编号为a和b的两个城邦,它们之间的桥如果要装饰起来,需要的费用如下计算:找到a和b在十进制下所有不同的数位,将数位上的数字求和。
例如,编号为2021和922两个城邦之间,千位、百位和个位都不同,将这些数位上的数字加起来是 (2+0+1) + (0+9+2) = 14。注意 922 没有千位,千位看成 0。
为了节约开支,政府准备只装饰 2020 座桥,并且要保证从任意一个城邦到任意另一个城邦之间可以完全只通过装饰的桥到达。
请问,小蓝国政府至少要花多少费用才能完成装饰。提示:建议使用计算机编程解决问题。
分析:并查集-Kruskal算法
#include <iostream>
#include <algorithm>
using namespace std;
int p[2030];
struct Edge
{
int a, b, w;
bool operator <(Edge const &t) const
{
return w < t.w;
}
}e[20000001];
int get_cost(int a, int b)
{
int sum = 0;
while (a || b)
{
if (a % 10 != b % 10)
{
sum += (a % 10 + b % 10);
}
a = a / 10;
b = b / 10;
}
return sum;
}
int find(int a)
{
return a == p[a] ? a : (p[a] = find(p[a]));
}
int main()
{
int k = 0;
for (int i = 1; i <= 2021; i++) p[i] = i;
for(int i = 1; i <= 2021; i++)
for (int j = i + 1; j <= 2021; j++)
{
int w;
w = get_cost(i, j);
e[k++] = { i, j, w };
}
sort(e, e + k);
int ans = 0;
for (int i = 0; i < k; i++)
{
int a = e[i].a, b = e[i].b;
if (find(a) != find(b))
{
ans += e[i].w;
p[find(a)] = find(b);
}
}
cout << ans << endl;
return 0;
}
结果:4046