jzoj 3810 Watering the Fields

Description

由于缺少雨水，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 计算他最少需要为连接他所有田地的管道网络支付多少钱。

---

Input

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

Output

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

---

Sample Input

3 11
0 2
5 0
4 3

Sample Output

46

样例解释：

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

Data Constraint

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

---

解题思路

很容易看出是最小生成树，不过限制了一些小于C的边不能选，在输入使就去掉这些边，然后做一次最小生成树就可以了。

---

Source Code

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

const int N=2001;

struct note
{
    int x,y,l;
};

int fa[N],a[N][2];
note side[N*N];
int n,c,lo;

void init(){
    scanf("%d%d",&n,&c);
    int i,j;
    for (i=1;i<=n;i++)
      scanf("%d%d",&a[i][0],&a[i][1]);
    lo=0; 
    for (i=1;i<=n;i++)
      for (j=i+1;j<=n;j++)
      {
        lo++;
        side[lo].x=i; side[lo].y=j;
        side[lo].l=(a[i][0]-a[j][0])*(a[i][0]-a[j][0])+(a[i][1]-a[j][1])*(a[i][1]-a[j][1]);
                if (side[lo].l<c) lo--;
      }
}

void qsort(int l,int r)
{
    int mid=side[(l+r)/2].l;
    int i=l,j=r;
    while (i<=j)
    {
        while ((side[i].l<mid) && (i<=r)) i++;
        while ((side[j].l>mid) && (j>=l)) j--;
        if (i<=j) {side[0]=side[i]; side[i]=side[j];side[j]=side[0]; i++;j--;}
    }
    if (l<j) qsort(l,j);
    if (i<r) qsort(i,r);
}

int root(int x)
{
    int i,ro;
    for (i=x;fa[i]!=0;i=fa[i]);
    ro=i;
    for (i=x;fa[i]!=0;i=fa[i]) fa[i]=ro;
    return ro;
}

int main()
{
    init();
    qsort(1,lo);
    memset(fa,0,sizeof(fa));
    int k=0,ans=0,j=0;
    while ((k!=n-1) && (j<lo))
    {
        j++;
        int u,w;
        u=root(side[j].x); w=root(side[j].y);
        if (u!=w) 
        {
            fa[u]=w;
            ans+=side[j].l;
            k++;
        }
    }
    if (k==n-1) printf("%d",ans);
      else printf("-1");
    return 0;
}