洛谷 P2872 [USACO07DEC]Building Roads S （题解+代码）

题目传送门：https://www.luogu.com.cn/problem/P2872
题解： 先计算所有点之间的距离，然后将所有给定边的边权值设为0。然后进行Kruskal算法即可。

代码及注释如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
#define two(x) (x)*(x)
using namespace std;
const int MAXN = 1e3+5;
const int MAXM = 1e6+1005;
struct node{
    int x,y;
    double z;
    friend bool operator < (node a,node b) {
        return a.z < b.z;//优先边权小的
    }
}a[MAXM];
int dis[MAXN];//记录祖先节点（是否已连接
double x[MAXN],y[MAXN];//记录坐标
double cal(int i,int j) {//计算i到j的距离
    return sqrt(two(x[i]-x[j])+two(y[i]-y[j]));
}
int getf(int p) {//计算p的祖先（路径压缩
    return dis[p]==p?p:dis[p]=dis[getf(dis[p])];
}
int main() {
    int n,m,cnt = 0;
    cin>>n>>m;
    for(int i = 1;i <= n;i++) {
        cin>>x[i]>>y[i];
        dis[i] = i;//初始化祖先为本身
    }
    for(int i = 1;i <= m;i++) {
        int x,y;
        cin>>x>>y;
        a[++cnt] = {x,y,0};//建边权为0的边
    }
    for(int i = 1;i <= n;i++) {//计算所有点之间的距离
        for(int j = 1;j <= n;j++) {
            if(i==j) continue;//跳过自身
            double val = cal(i,j);//计算i到j的距离
            a[++cnt] = {i,j,val};//添加边
        }
    }
    sort(a+1,a+cnt+1);//排序
    double ans = 0;//计算添加的边权和
    for(int i = 1;i <= cnt;i++) {
        node tmp = a[i];
        if(dis[getf(tmp.x)]!=dis[getf(tmp.y)]) {//若两点还未连接
            ans += tmp.z;//添加边权值
            dis[getf(tmp.x)] = dis[getf(tmp.y)];//向左合并
        }
    }
    printf("%.2f",ans);
    return 0;
}