10分Kruskal蒟蒻求助
查看原帖
10分Kruskal蒟蒻求助
205782
R浩轩泽Anmicius楼主2020/7/2 17:34

这两天在最小生成树上栽了跟头,不管是 P2330 繁忙的都市 还是 P2872 Building Roads S 两道Kruskal都只有十分......望大佬解惑

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 1005
#define E 1000005
struct node{
	int x;
	int y;
}p[N];
struct Edge{
	int x;
	int y;
	int val;
}edge[E];
int n,m,num,fa[N],k;
double ans;
int find(int x)//找根、压缩路径
{
	if(fa[x]!=x)fa[x]=find(fa[x]);
	return fa[x];
} 
void unionn(int x,int y)
{
	x=find(x),y=find(y);
	fa[y]=x;
}
bool cmp(Edge a,Edge b)
{
	return a.val<b.val;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=n;i++)
	scanf("%d%d",&p[i].x,&p[i].y);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		unionn(a,b);
	}
	for(int i=1;i<n;i++)
	for(int j=i+1;j<=n;j++)
	edge[++num].val=sqrt((pow((p[i].x-p[j].x),2)+pow((p[i].y-p[j].y),2))*1.00),edge[num].x=i,edge[num].y=j;
	sort(edge+1,edge+1+num,cmp);
	for(int i=1;i<=num;i++)//枚举边 
	{
		int x=edge[i].x,y=edge[i].y;
		if(find(x)!=find(y))//不要用fa[]比较
		{
			unionn(x,y);
			ans+=edge[i].val*1.00;
			k++;
			if(k==(n-1))break;
		}
	}
	printf("%.2lf\n",ans);
	return 0;
}

奉上P2872的代码

2020/7/2 17:34
加载中...