题解 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;
}