最小生成树 prim算法（附P1265 题解）

最小生成树 prim算法（附PP1265 题解）

最小生成树prim算法原理

prim算法也使用了dist数组，但是和迪杰斯特拉这些算法中dist含义不同，prim算法中的dist算法是指各个点到连通部分的最短距离

首先随机选取一个点加入连通块，并利用这个点初始化其他点到连通块距离dist[j] = min(dist[j],g[t][j]),然后后续选取不在连通块中，但是距离连通块最近的点作为t，继续更新其他点到连通块距离

题目

公路修建

题目描述

某国有 $n$ 个城市，它们互相之间没有公路相通，因此交通十分不便。为解决这一“行路难”的问题，政府决定修建公路。修建公路的任务由各城市共同完成。

修建工程分若干轮完成。在每一轮中，每个城市选择一个与它最近的城市，申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。

政府审批的规则如下：

1. 如果两个或以上城市申请修建同一条公路，则让它们共同修建；
2. 如果三个或以上的城市申请修建的公路成环。如下图，A 申请修建公路 AB，B 申请修建公路 BC，C 申请修建公路 CA。则政府将否决其中最短的一条公路的修建申请；
3. 其他情况的申请一律同意。

一轮修建结束后，可能会有若干城市可以通过公路直接或间接相连。这些可以互相连通的城市即组成“城市联盟”。在下一轮修建中，每个“城市联盟”将被看作一个城市，发挥一个城市的作用。

当所有城市被组合成一个“城市联盟”时，修建工程也就完成了。

你的任务是根据城市的分布和前面讲到的规则，计算出将要修建的公路总长度。

输入格式

第一行一个整数 $n$，表示城市的数量。（$n\le 5000$）

以下 $n$ 行，每行两个整数 $x$ 和 $y$，表示一个城市的坐标。（$-10^6\le x,y\le 10^6$）

输出格式

一个实数，四舍五入保留两位小数，表示公路总长。（保证有唯一解）

样例 #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;
}