计蒜客习题：连线问题

问题描述

蒜头君和花椰菜君经常出难题考对方。一天，花椰菜君给蒜头君出了这样一道难题：花椰菜君在坐标系上随机画了 N 个点，然后让蒜头君给点之间连线，要求任意两点之间都是连通的，且所连的线段长度之和最小。聪明的你快来帮蒜头君解决一下吧。

输入格式

第一行输入一个整数 N(1≤N≤100)，表示花椰菜君一共画了 N 个点。然后输入 N 行，每行输入两个整数 x,y（0≤x,y≤1,000），代表一个点的坐标。

输出格式

输出一行，输出一个实数，表示所连线段长度之和的最小值（结果四舍五入保留两位小数）。

样例输入

4

1 1

2 2

3 3

3 1

样例输出

4.24

AC代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
int INF=0x3f3f3f3f;
int dad[500];
int G[5000][2];

struct edge{
int u,v;
double w;
}e[200000];

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

int getdad(int i)
{
if(dad[i]==i)return dad[i];
return dad[i]=getdad(dad[i]);
}
int main() {
int n;
for (int i = 1; i <= 105; i++)
dad[i] = i;
memset(G, INF, sizeof(G));
cin >> n;
for(int i=1;i<=n;i++)
{
cin>>G[i][0]>>G[i][1];
}
int eid = 1;
for (int i = 1; i <= n; i++) {
for (int j = i + 1; j <= n; j++) {
e[eid].u = i;
e[eid].v = j;
e[eid].w = sqrt((G[i][0]-G[j][0])*(G[i][0]-G[j][0])+(G[i][1]-G[j][1])*(G[i][1]-G[j][1]));
eid++;
}
}
sort(e+1, e+eid, cmp);
int rest = n;
double ans = 0;
for (int i
b1ec
= 1; i < eid - 1 && rest > 1; i++) {
int x = e[i].u;
int y = e[i].v;
if (getdad(x) != getdad(y)) {
dad[getdad(x)] = getdad(y);
ans += e[i].w;
rest--;
}
}
//ans+=0.005;
printf("%.2f", ans);
return 0;
}