2603 公路修建

题目描述 Description

某国有n个城市，它们互相之间没有公路相通，因此交通十分不便。为解决这一"行路难"的问题，政府决定修建公路，修建公路的任务由各城市共同完成。

修建工程分若干轮完成。在每一轮中，每个城市选择一个与它最近的城市，申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。

政府审批的规则如下：

(1)如果两个或以上城市申请修建同一条公路，则让它们共同修建；

(2)如果三个或以上的城市申请修建的公路成环。如，A申请修建公路AB，B申请修建公路BC，C申请修建公路CA，则政府将否决其中最短的一条公路的修建申请；

 (3)其他情况的申请一律同意。

一轮修建结束后，可能会有若干城市可以通过公路直接或间接相连。这些可以互相：连通的城市即组成"城市联盟"。在下一轮修建中，每个"城市联盟"将被看作一个城市，发挥一个城市的作用。

当所有城市被组合成一个"城市联盟"时，修建工程也就完成了。

你的任务是根据城市的分布和前面讲到的规则，计算出将要修建的公路总长度。

输入描述 Input Description

第一行一个整数n(n≤5000),表示城市的数量。

以下n行，每行两个整数x和y(-106≤x、y≤106)，表示一个城市的坐标。

思路：由于点比较多，所以只能用prim

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define IOS ios::sync_with_stdio(false)
#define pa pair<double,int>
#define ll long long
using namespace std;
struct edge {
    int to, next;
} e[25000005];
int n, cnt, last[5005];ll x[5005], y[5005], vis[5005];
double ans, dis[5005];
void ins(int u, int v) {
    e[++cnt].to = v;
    e[cnt].next = last[u];
    last[u] = cnt;
}
double len(int a, int b) {
    return sqrt((x[a] - x[b]) * (x[a] - x[b]) + (y[a] - y[b]) * (y[a] - y[b]));
}
void prim() {
    priority_queue<pa, vector<pa>, greater<pa> > q;
    for(int i = 1; i <= n; i++)
        dis[i] = INF;
    dis[1] = 0;
    q.push(make_pair(0, 1));
    while(!q.empty()) {
        int now = q.top().second;
        q.pop();
        if(vis[now])
            continue;
        vis[now] = 1;
        ans += dis[now];
        for(int i = last[now]; i; i = e[i].next) {
            int to = e[i].to;
            double v = len(now, to);
            if(v < dis[to]) {
                dis[to] = v;
                q.push(make_pair(dis[to], to));
            }
        }
    }
}
int main() {
    IOS;
    cin >> n;
    for(int i = 1; i <= n; i++) {
        cin >> x[i] >> y[i];
    }
    for(int i = 1; i <= n; i++)
        for(int j = i + 1; j <= n; j++)
            ins(i, j), ins(j, i);
    prim();
    printf("%.2f", ans);
    return 0;
}