计蒜客习题：连线问题

---

问题描述

蒜头君和花椰菜君经常出难题考对方。一天，花椰菜君给蒜头君出了这样一道难题：花椰菜君在坐标系上随机画了 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 = 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;
}