[洛谷] P4047 部落划分：最小生成树

一. 题目描述

二.解题思路

呜呜呜是我太菜了，大家都说这是一道水题，可是我还是看来别人的解答才自己写出来。具体的解题思路可以看这一篇。使用Kruskal算法。初始时有n个居住点就有n个部落。每次都选最短边，每选一条，部落数就-1。最后需要k个部落，那么就选择了n-k条边，所以第n-k+1边的长度即为所求。

三. AC代码

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
using namespace std;

const int MAX = 1010;
int n, k, index; //n个点，k个部落
int x[MAX], y[MAX];

struct edge {
    int u;
    int v;
    double cost;
}E[1000010];

double d[MAX];
int father[MAX];
int findFather(int x) {
    while (x != father[x]) {
        x = father[x];
    }
    return x;
}

bool cmp(edge a, edge b) {
    return a.cost < b.cost;
}

double kruskal() {
    int num = 0;
    int j = 0;
    sort(E, E + index, cmp);
    for (int i = 1; i <= n; i++) {
        father[i] = i;
    }
    for (int i = 0; i < index; i++) {
        int fau = findFather(E[i].u);
        int fav = findFather(E[i].v);
        if (fau != fav) {
            father[fau] = fav;
            d[j++] = E[i].cost;
            if (j == n - k + 1) {
                return d[j - 1];
            }
        }
    }
    return 0.0;
}

int main() {
    cin >> n >> k;
    for (int i = 0; i < n; i++) {
        cin >> x[i] >> y[i];
    }

    for (int i = 0; i < n; i++) {
        for (int j = i + 1; j < n; j++) {
            double distance = sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j])); //计算两点之间的距离
            E[index++] = { i,j,distance };
        }
    }
    double res = kruskal();
    printf("%.2lf\n", res);
    return 0;
}