写在前面
这是一道蓝桥杯省模拟题因为我还不会动态规划解决这个问题 所以模拟的时候我直接暴力解决,可能会突破时间限制,等以后对动归很熟悉了,在更新这篇文章,以下是我当时在模拟的时候暴力解决的算法。如果代码有问题,请私信于我,共同探讨,共同进步。
问题描述
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。
算法思路
先用一个二维数组记录点到点之间的所要花的钱,然后全排列这几个村庄的序号,在全排列的时候,进行最小值计算,最后记得保留小数点后两位。这个算法空间上有很大的优化,时间上也是,等以后学习多了,再回来更新、感谢你的阅读。
代码实现
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
double x[1000];
double y[1000];
double h[1000];
double array1[100][100];
int max[1000];
int a[1000];
int main()
{
int n = 0;
cin >> n;
for (int i = 0; i < n; ++i)
{
cin >> x[i];
cin >> y[i];
cin >> h[i];
}
//int *array =(int*)malloc(sizeof(int)*n*(n-1));
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < n; ++j)
{
if (i == j)
{
array1[i][j] = -1;//如果i j 相同不考虑这种情况则置位-1
continue;
}
//在数组中存放每一个村庄对应各个村庄的值 i j 表示i村庄和j村庄连通所要花的钱
array1[i][j] = sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j])) + (h[i] - h[j]) * (h[i] - h[j]);
}
}
for (int i = 0; i < n; i++)
{
a[i] = i;//有多少个村庄就存入多少个数组
}
double maxsum = 1000000;
double sum = 0;
do
{
for (int i = 0; i < n - 1; ++i)
{
sum += array1[a[i]][a[i + 1]]; //计算每一种组合所要花的钱
//cout<<a[i]<<" ";
}
if (sum < maxsum) //记录最小值
{
maxsum = sum;
}
//cout<<endl;
sum = 0; //归零进行下一种情况的排列组合
} while (next_permutation(a, a + n));//将n种村庄进行排列组合
printf("%.2lf", maxsum);
return 0;
}