最小生成树 prim算法(附PP1265 题解)
最小生成树prim算法原理
prim算法也使用了dist数组,但是和迪杰斯特拉这些算法中dist含义不同,prim算法中的dist算法是指各个点到连通部分的最短距离
首先随机选取一个点加入连通块,并利用这个点初始化其他点到连通块距离dist[j] = min(dist[j],g[t][j])
,然后后续选取不在连通块中,但是距离连通块最近的点作为t,继续更新其他点到连通块距离
题目
公路修建
题目描述
某国有 n n n 个城市,它们互相之间没有公路相通,因此交通十分不便。为解决这一“行路难”的问题,政府决定修建公路。修建公路的任务由各城市共同完成。
修建工程分若干轮完成。在每一轮中,每个城市选择一个与它最近的城市,申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。
政府审批的规则如下:
- 如果两个或以上城市申请修建同一条公路,则让它们共同修建;
- 如果三个或以上的城市申请修建的公路成环。如下图,A 申请修建公路 AB,B 申请修建公路 BC,C 申请修建公路 CA。则政府将否决其中最短的一条公路的修建申请;
- 其他情况的申请一律同意。
一轮修建结束后,可能会有若干城市可以通过公路直接或间接相连。这些可以互相连通的城市即组成“城市联盟”。在下一轮修建中,每个“城市联盟”将被看作一个城市,发挥一个城市的作用。
当所有城市被组合成一个“城市联盟”时,修建工程也就完成了。
你的任务是根据城市的分布和前面讲到的规则,计算出将要修建的公路总长度。
输入格式
第一行一个整数 n n n,表示城市的数量。( n ≤ 5000 n \leq 5000 n≤5000)
以下 n n n 行,每行两个整数 x x x 和 y y y,表示一个城市的坐标。( − 1 0 6 ≤ x , y ≤ 1 0 6 -10^6 \leq x,y \leq 10^6 −106≤x,y≤106)
输出格式
一个实数,四舍五入保留两位小数,表示公路总长。(保证有唯一解)
样例 #1
样例输入 #1
4
0 0
1 2
-1 2
0 4
样例输出 #1
6.47
提示
修建的公路如图所示:
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 5010;
typedef struct node {
double x, y;
} Node;
Node mynode[N];
double dist[N];
bool visited[N];
double getDistance(double x1, double y1, double x2, double y2) {
return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
double prime() {
memset(visited, false, sizeof(visited));
for (int i = 1; i <= n; i++) {
dist[i] = 1e12; // 将距离数组初始化为一个较大的值
}
dist[1] = 0; // 初始节点的距离设为0
double res = 0;
for (int i = 1; i <= n; i++) {
int u = -1;
for (int j = 1; j <= n; j++) {
if (!visited[j] && (u == -1 || dist[j] < dist[u])) {
u = j;
}
}
if (dist[u] == 1e12) return 1e12; // 如果存在无法到达的节点,返回一个较大的值
visited[u] = true;
res += dist[u];
for (int v = 1; v <= n; v++) {
if (!visited[v]) {
double d = getDistance(mynode[u].x, mynode[u].y, mynode[v].x, mynode[v].y);
dist[v] = min(dist[v], d); // 更新距离数组
}
}
}
return res;
}
int main() {
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> mynode[i].x >> mynode[i].y;
}
double ans = prime();
printf("%.2lf\n", ans);
return 0;
}