P1265 公路修建（洛谷）（prim最小生成树）

两个最小生成树算法, 都有一个共同的思想: 这棵树是一点一点长大的; 并且每次生长, 都是贪心的.
不同之处是:
Kruscal算法是以边为中心的, 每次找最小的并且有用的边添加到树上;
Prim算法是以点为中心的, 每次找离树最近的点添加到树上.
我们可以把一棵树理解成一个有智能的生命, 可以感知它附近的点到它的距离. 每次生长枝条, 它都选择离它最近的那个点.
点到树的距离, 是指树外一个点到树上的任意点的最小距离.
所以,在代码实现的时候, 需要维护这样一个数组: 树外的点到树的距离. 所以, 还需要区分一下点究竟在树上还在树外.
维护数组就是要做两件事: 更改数组和调用数组.
何时更改: 树外的点到树的距离发生变化. 这种事只能在树生长了新的枝条的时候发生, 因为新加入的那个点可能更新了树到树外的某个点的距离. 同时, 新加入的那个点变成了树上的点.
何时调用: 想扩张的时候, 找最近的树外点.
一开始, 我们认为1号结点在树上, 把它到树的距离设为0; 其它结点的距离是INF.
在调用完整个算法之后, 因为当一个节点变成树上节点后, 我们就不再更新它到树的距离, 所以, 原来的数组记录的就是它加入到树上时的连边长度. 对之求和, 就是树上所有边权之和.
Prim算法的优势: 稠密图, 尤其是完全图.
因为在Kurscal算法中, 必须事先求出所有边的长度才能对之排序. 但是一个有5000节点的完全图, 这样做的空间开销是巨大的.
Prim算法只在更新点到树的距离时需要用到边长, 因此对于这种给坐标的完全图, 可以现用现算.

P1265 公路修建 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include <iostream>
#include <bits/stdc++.h>
#include <algorithm>
#include <sstream>
#include <unordered_map>
#include <set>
#include <queue>
#include <deque>
#include <map>
#include <string>
#include <cstring>

#define x first
#define y second
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef pair<int,int> PII;
typedef pair<char,int> PCI;
typedef long long LL;
typedef unsigned long long ULL;

const int N=5010, INF = 0x3f3f3f3f ;



struct node{
    LL x,y;
    LL operator *(const node &W) const
    {
        return (x-W.x) *(x - W.x) + (y-W.y) * (y -W.y );
    }
}q[N];

int n,m;
LL d[N];
bool st[N];


int prim()
{
    memset(d,0x3f,sizeof d);
    memset(st,0,sizeof st);
    d[1] =0; // 选第一个点为 根节点  
    
    for(int i=1;i<n;i ++ )
    {
        int t =-1;
        for(int j=1;j<=n;j ++ )
        if(!st[j] && (t ==- 1 || d[t] > d[j])) 
        t =j;
        
        st[t] =1;
        for(int j=1;j<=n;j ++ )
        if(!st[j]) d[j] =min(d[j],q[t] * q[j]);
        // 更新 每个未被处理的 点到当前点的 最短距离  
    }
    
    
    
    
    
}

void solve()
{
    cin>>n;
    for(int i=1;i<=n ;i ++ )
    cin>>q[i].x >> q[i].y ;
    
    prim();
    double ans =0;
    for(int i=1;i<=n;i ++ ) ans += sqrt((double)d[i]);
    
    printf("%.2lf\n",ans);
}


int main()
{
    ios 
    int T=1;
//    cin>>T;
    while(T -- )
    {
        solve();
    }    
    
    
    
    
    return 0;
}