2017.1.14【初中部 GDKOI】模拟赛B组 Watering the Fields 题解

原题：

http://172.16.0.132/senior/#contest/show/1895/0

题目描述：

由于缺少雨水，FJ 想要建造一个在他的N（1 N 2000）块田之间送水的灌溉系统。
每块田地i 由一个二维平面上独一无二的点(xi; yi) 描述，这里有0 <= xi; yi <=1000。在两块田地i 和j 之间建造水管的费用等于它们之间欧几里得距离的平方：(xi-xj)^2+(yi-yj)^2
FJ 希望建造一个连接所有田地并且花费最小的管道系统——满足从任意田地出发，水可以通过一系列管道到达另外的任意一块田地。
不幸的是，帮助FJ 安装灌溉系统的承包商拒绝安装任何花费（欧几里得长度平方）小于C（1 C 1; 000; 000）的管道。
请帮助FJ 计算他最少需要为连接他所有田地的管道网络支付多少钱。

输入：

第一行：整数N 和C。
第2 至N + 1 行：第i 行包含整数xi 和yi。

输出：

输出单独一行一个整数——连接所有田地的管道网络的最小费用，或者当满足条件的网络不可能建造时输出-1。

样例输入：

3 11
0 2
5 0
4 3

样例输出：

46

样例解释：

有三块田地，分别在坐标(0; 2); (5; 0) 和(4; 3)。承包商将只会安装费用不少于11 的管道。
FJ 不能建造连接分别在(4; 3) 和(5; 0) 的田地的管道，因为它的费用只有10。
故他只得建造连接(0; 2) 和(5; 0) 且费用为29 的管道，以及连接(0; 2) 和(4; 3)且费用为17 的管道。

数据范围限制：

对于20% 的数据，有N <= 10。
对于30% 的数据，有N <= 100。
对于40% 的数据，有N <= 200。

分析：

裸最小生成树，优先队列优化prim。

实现：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;

struct id1
{
    int x,y;
};
struct id2
{
    int fr,to,co;
};
int fa[2005],tot,n,c,i,j,v,x,y,cnt;
long long ans;
id1 w[2005];
id2 a[2005*2005],e;
bool cmp(id2 a,id2 b)
{
    return a.co<b.co;
}
int find(int a)
{
    if(fa[a]!=a)fa[a]=find(fa[a]);
    return fa[a];
}
int main()
{
    scanf("%d %d",&n,&c);
    for(i=1;i<=n;i++)
    {
        scanf("%d %d",&w[i].x,&w[i].y);
        fa[i]=i;
    }
    for(i=1;i<=n;i++)
        for(j=i+1;j<=n;j++)
        {
            v=(w[i].x-w[j].x)*(w[i].x-w[j].x)+(w[i].y-w[j].y)*(w[i].y-w[j].y);
            if(v<c)continue;
            a[++tot]=(id2){i,j,v};
        }
    sort(a+1,a+tot+1,cmp);
    for(i=1;i<=tot;i++)
    {
        e=a[i]; x=find(e.fr); y=find(e.to);
        if(x!=y)
        {
            fa[y]=x;
            ans+=e.co;
        }
    }
    for(i=1;i<=n;i++)
        if(fa[i]==i) cnt++;
    if(cnt>1) printf("-1");
    else printf("%lld",ans);
}