题解 P2212 【[USACO14MAR]浇地Watering the Fields】

裸的一个最小生成树

在每次进行加边的时候判断加入的边是否合法 同时更新加入的边的数目

对一个n个节点的图而言 它的最小生成树的的节点数目应该是n-1

如果加入的边数小于n-1 说明无解 输出-1

代码如下：

#include<bits/stdc++.h>
using namespace std;
int n,c,tot,px,py,f[2005],l,cnt,ans;
struct p1{
    int x,y;
}a[2005];
struct p2{
    int x,y,z;
}b[4000005];
int dis(int x1,int y1,int x2,int y2){
    return (x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
}
int find(int x){
    if (x==f[x]) return x;
    return f[x]=find(f[x]);
}
bool cmp(p2 a,p2 b){
    return a.z<b.z;
}
int main(){
    scanf("%d%d",&n,&c);
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=n;i++) scanf("%d%d",&a[i].x,&a[i].y);
    for (int i=1;i<=n;i++)
        for (int j=i+1;j<=n;j++){
            l=dis(a[i].x,a[i].y,a[j].x,a[j].y);
            tot++;
            b[tot].x=i;
            b[tot].y=j;
            b[tot].z=l;
        }
    sort(b+1,b+tot+1,cmp);
    for (int i=1;i<=tot;i++){
        if (b[i].z<c) continue;
        px=find(b[i].x);
        py=find(b[i].y);
        if (px==py) continue;
        f[px]=py;
        ans+=b[i].z;
        cnt++;
    }
    if(cnt!=n-1) printf("-1\n");
        else printf("%d\n",ans);
    return  0;
}