【蓝桥杯】2020-04省赛模拟 A组 问题十 村庄通电

第十题

问题描述：
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。

个人思路

用结构体记录输入的村庄x、y、h信息，把题目要求的价格计算方法作为村庄（作为图的顶点）之间的权值，利用prim算法求出最小生成树。

参考代码（有误请指正）

# include <iostream>
# include <math.h>
# include <iomanip>
using namespace std;
const int MAX_SIZE = 1004;
const int INF = 99999;
struct vli{
    int x,y,h;
};
vli v[MAX_SIZE];
double lowcost[MAX_SIZE] = {0};
double dis[MAX_SIZE][MAX_SIZE];
//计算耗费
double value(vli a,vli b){
    double dis = sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)) + (a.h-b.h)*(a.h-b.h);
    return dis;
}
//找出最小边
int findmin(int n){
    int min = 1;
    for(int i=2;i<=n;i++)
        if(lowcost[i]<lowcost[min])
            min = i;
    return min;
}
double prim(int n){
    double sum = 0;
    for(int i=1;i<=n;i++)
        lowcost[i] = dis[1][i];
    for(int i=2;i<=n;i++){
        int j = findmin(n);
        sum += lowcost[j];
        lowcost[j] = INF;
        for(int k=1;k<=n;k++){//更新lowcost
            if(dis[j][k]<lowcost[k] && lowcost[k]!=INF)
                lowcost[k] = dis[j][k];
        }
    }
    return sum;
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>v[i].x>>v[i].y>>v[i].h;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            if(i==j)
                dis[i][j] = dis[j][i] = INF;
            else
                dis[i][j] = dis[j][i] = value(v[i],v[j]);
        }
    cout<<setiosflags(ios::fixed)<<setprecision(2)<<prim(n);
    return 0;
}