题解 | #Freckles#

Freckles

https://www.nowcoder.com/practice/41b14b4cd0e5448fb071743e504063cf

//这题注意距离计算和小数的问题即可
#include <iostream>
#include<cstring>
#include<string>
#include<vector>
#include<cmath>
#include<algorithm>
const int maxn = 110;
using namespace std;
int father[maxn];
int height[maxn];

struct Edge {
    int from;
    int to;
    double length;
    bool operator<(const Edge& e) {
        return length < e.length;
    }
};

struct Point {
    double x, y;
    Point(double x, double y): x(x), y(y) {}
};

double distance1(Point* p1, Point* p2) {
    double dis = (p1->x - p2->x) * (p1->x - p2->x) + (p1->y - p2->y) *
                 (p1->y - p2->y);
    return sqrt(dis);
}

Edge edge[maxn * maxn];

void init(int n) {
    for (int i = 0; i <= n; i++) {
        father[i] = i;
        height[i] = 0;
    }
}

int find(int x) {
    if (x != father[x]) {
        father[x] = find(father[x]);
    }
    return father[x];
}

void Union(int x, int  y) {
    x = find(x);
    y = find(y);
    if (x != y) {
        if (height[x] < height[y]) {
            father[x] = y;
        } else if (height[x] > height[y]) {
            father[y] = x;
        } else {
            father[y] = x;
            height[x] += 1;
        }
    }
}

double kruskal(int n, int edgeNumber) {
    init(n);
    sort(edge, edge + edgeNumber);
    double sum = 0;
    for (int i = 0; i < edgeNumber; i++) {
        Edge current = edge[i];

        if (find(current.from) != find(
                    current.to)) { //将这条边联通不会产生回环

            Union(current.from, current.to);
            sum += current.length;
        }
    }
    return sum;

}

//3
//1.0 1.0
//2.0 2.0
//2.0 4.0

int main() {
    int n;
    while (cin >> n) {
        if (n == 0) break;
        double x, y;
        vector<Point*> points;
        for (int i = 0; i < n; i++) {
            cin >> x >> y;
            points.push_back(new Point(x, y));
        }
        int num = 0;
        for (int i = 0; i < n - 1; i++) {
            for (int j = i + 1; j < n; j++) {
                edge[num].from = i;
                edge[num].to = j;
                edge[num].length = distance1(points[i], points[j]);
                num++;
            }
        }

        printf("%0.2f\n", kruskal(n, num));


    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务