题目描述:
题意:给出n个点的坐标,找出最短路。
Input:
第一行包含0 < n <= 100,点的数目。每一个点都有一条线;下面每一行都有两个实数,表示该点的(x,y)坐标。输入包含多个测试案例
Output:
你的程序打印一个实数到小数点后两位:可以连接所有点的最小墨线总长度。
输入:
3
1.0 1.0
2.0 2.0
2.0 4.0
输出:
3.41
思路:记录下每两个点的距离,在进行排序后,利用并查集,把所有的点连接起来距离之和即为最终结果。
代码:
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
#define ll long long
const ll N = 1e4 + 5;
ll pre[105];
struct node {
ll a, b;
double dis;
} p[N];
double x[105];
double y[105];
int n;
ll find(ll x) {
if (x == pre[x])return pre[x];
return pre[x] = find(pre[x]);
}
int cmp(node a, node b) {
return a.dis < b.dis;
}
int main() {
while (cin >> n) {
double ans = 0;
ll cnt = 1;
for (int i = 1; i <= n; i++) {
cin >> x[i] >> y[i];
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
p[cnt].a = i;
p[cnt].b = j;
p[cnt].dis = sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]));
cnt++;
}
}
for (int i = 1; i <= n; i++)pre[i] = i;
sort(p + 1, p + cnt + 1, cmp);
for (int i = 1; i < cnt; i++) {
ll g = find(p[i].a);
ll h = find(p[i].b);
if (g != h) {
pre[g] = h;
ans += p[i].dis;
}
}
printf("%.2lf\n", ans);
}
return 0;
}