蓝桥杯模拟赛2021最小生成树-通电-模板题

题目

J 题:
问题描述
　　2015年，全中国实现了户户通电。作为一名电力建设者，小明正在帮助一带一路上的国家通电。
　　这一次，小明要帮助 n 个村庄通电，其中 1 号村庄正好可以建立一个发电站，所发的电足够所有村庄使用。
　　现在，这 n 个村庄之间都没有电线相连，小明主要要做的是架设电线连接这些村庄，使得所有村庄都直接或间接的与发电站相通。
　　小明测量了所有村庄的位置（坐标）和高度，如果要连接两个村庄，小明需要花费两个村庄之间的坐标距离加上高度差的平方，形式化描述为坐标为 (x_1, y_1) 高度为 h_1 的村庄与坐标为 (x_2, y_2) 高度为 h_2 的村庄之间连接的费用为
　　sqrt((x_1-x_2)(x_1-x_2)+(y_1-y_2)(y_1-y_2))+(h_1-h_2)*(h_1-h_2)。
　　在上式中 sqrt 表示取括号内的平方根。请注意括号的位置，高度的计算方式与横纵坐标的计算方式不同。
　　由于经费有限，请帮助小明计算他至少要花费多少费用才能使这 n 个村庄都通电。
输入格式
　　输入的第一行包含一个整数 n ，表示村庄的数量。
　　接下来 n 行，每个三个整数 x, y, h，分别表示一个村庄的横、纵坐标和高度，其中第一个村庄可以建立发电站。
输出格式
　　输出一行，包含一个实数，四舍五入保留 2 位小数，表示答案。
样例输入
4
1 1 3
9 9 7
8 8 6
4 5 4
样例输出
17.41
评测用例规模与约定
　　对于 30% 的评测用例，1 <= n <= 10；
　　对于 60% 的评测用例，1 <= n <= 100；
　　对于所有评测用例，1 <= n <= 1000，0 <= x, y, h <= 10000。

分析
最小生成树, 最多1000个点, 也就是说最多也就 1000 * (1000 - 1) / 2 条无向边, 可以抽象成n个顶点，生成的一张无向完全图，然后找到该图的一个最小生成树，权值的计算方法也给出来了，因此可以考虑使用prim 或 kruscal算法解决，本人觉得kruscal的逻辑比较清晰，所以选择了使用kruscal算法解决这个问题，喜欢prim的同学也可以使用此算法解决（解不了所有数据）。
e<=nx(n-1)/2=100x99/2=4950;
kruscal的时间复杂度是 18,288.3 =4950x 3.69 =O（eloge）<100W<O（e 2 ）=O( 4950^2)
1S时间可以处理100W数据
18288.3<<100W
所以kruscal可以处理题目范围内的所有数据
(sort忘记改成堆排序了 sort_heap)

模板如下：

#include <iostream>
#include<algorithm>
using namespace std;
const int N=1000;
/* run this program using the console pauser or add your own getch, system("pause") or input loop */

struct EDGE{
	int u,v,cost;
	void set(int a,int b,int w){
		u=a;
		v=b;
		cost=w;
	}
};

EDGE edge[N*N-1/2];

bool cmp(EDGE a, EDGE b){
	return a.cost<b.cost;
}

int father[N];

int find(int x){
	if(father[x]==-1){
		return x;
	}else{
		return father[x]=find(father[x]);
	}
} 

bool union1(int x ,int y){
	x=find(x);
	y=find(y);
	if(x==y){
		return false;
	}
	if(x>y){
		father[x]=y;
	}else{
		father[y]=x;
	}
	return true;
}

int main(int argc, char *argv[]) {
	int i;
	int v,w,cost;
	cout<<"请输入边数和顶点数:"<<endl;
	int m,t;
	cin>>m>>t;	
	cout<<"请输入边的信息 例如 1 2 10："<<endl; 
	for(i=0;i<m;i++){
		cin>>v>>w>>cost;
		edge[i].set(v,w,cost);
	}
	sort(edge,edge+m,cmp);
	i=0;
	fill(father,father+m,-1);
	int sum=0,count=0;
	while(count!=t-1){
		if(union1(edge[i].u,edge[i].v)){
			cout<<edge[i].u<<"-"<<edge[i].v<<endl;
			sum+=edge[i].cost;
			count++;
		}
		i++;
	}
	cout<<"sum:"<<sum <<endl;
	return 0;
}

---

求点赞👍👍👍
原创不易，点赞容易。
您的鼓励就是我的最大动力！！！。

本篇博客到此结束，谢谢大家观看。