[洛谷]P2212 [USACO14MAR] Watering the Fields S

最小生成树一眼出，拿捏（进步啦）

几乎是模板了，只要让两点距离>=c的入“队”即可，然后跑最小生成树（kruskal）就好了

ACcode:

#include<bits/stdc++.h>
using namespace std;
const int N=2e3+10;
struct node{
	int x,y;
}a[N];
struct E{
	int u,v,w;
}e[N*N];
int n,c,t,cnt,fa[N],flag,ans;
bool cmp(E i,E j){
	return i.w<j.w;
}
int find(int x){
	return  fa[x]==x?x:fa[x]=find(fa[x]);
}
void kruskal(){
	for(int i=1;i<=t;i++){
		int f1=find(e[i].u);
		int f2=find(e[i].v);
		if(f1!=f2){
			fa[f1]=f2;
			ans+=e[i].w;
			cnt++;
		}
		if(cnt==n-1){
			flag=1;
			break;
		}
	}
}
void solve() {
    cin>>n>>c;
    for(int i=1;i<=n;i++){
    	cin>>a[i].x>>a[i].y;
	}
	for(int i=1;i<n;i++){
		for(int j=i+1;j<=n;j++){
			int len=(a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y);
			if(len>=c)
			e[++t]={i,j,len};
		}
	}
	for(int i=1;i<=n;i++) fa[i]=i;
	sort(e+1,e+1+t,cmp);
	kruskal();
	if(flag) cout<<ans<<"\n";
	else cout<<"-1"<<"\n";
}
int main() {

	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
	return 0;
}

over~