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