【最小生成树】P2212 [USACO14MAR]Watering the Fields S

大于等于c的建边,跑kruskal即可

#include<bits/stdc++.h>
using namespace std;
const int maxn=2005;
int n,c,tot;
int x[maxn],y[maxn],f[maxn];
struct edge
{
	int u,v,w;
}e[maxn*maxn];
bool cmp(edge aa ,edge bb)
{
	return aa.w<bb.w;
}
int find(int x)
{
	if(x==f[x]) return x;
	return f[x]=find(f[x]);
}
void kruskal()
{
	int ans=0,cnt=0;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=tot;i++)
	{
		int fa=find(e[i].u);
		int fb=find(e[i].v);
		if(fa==fb) continue;
		f[fa]=f[fb];
		ans+=e[i].w;
		cnt++;
		if(cnt==n-1) break;
	}
	if(cnt==n-1) printf("%d\n",ans);
	else printf("-1\n");
	return;
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	scanf("%d%d",&n,&c);
	for(int i=1;i<=n;i++)
		scanf("%d%d",&x[i],&y[i]);
	for(int i=1;i<=n;i++)
		for(int j=i+1;j<=n;j++)
		{
			int dis=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
			if(dis>=c) 
			{
				e[++tot].u=i; e[tot].v=j; e[tot].w=dis;
			}
		}
	sort(e+1,e+tot+1,cmp);
	kruskal();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/andyc_03/article/details/107743714