本文主要是介绍HDU 4463--最小生成树+并查集标记,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题意:
有n家店,要你把他们连在一起(即建成一颗最小生成树),耐克和苹果是要求必须有边直接连的,然后就给你n个店的坐标。输出你建好的生成树的权值。
输入:
4 2 3 0 0 1 0 0 -1 1 -1 0
输出:
3.41
分析:
最小生成树,首先将各个商店连接起来,将边进行排序,找出最小的边的权值,为了排序我们建立一个结构体,用来储存坐标,和边权,解决完排序问题之后,下面就是判断两个点是不是来自同一个子集,由于已经选择的边会将与该边连接的点归并到一个集合中,因此用并查集就可以代替标记数组,将不在同一个集合中的两个点,用最小的没有选择的边连接起来,这就是克鲁斯卡尔算法的实现方式。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <map>
#include <vector>
#include <queue>
using namespace std;int pa[100],dist1[100],dist2[100];
struct node
{int xx,yy; //xx是边的起始点,yy是边的终点double w; //边权
} a[10000];bool cmp(node a,node b)
{return a.w<b.w;
}int find(int x) //查找根节点
{return x==pa[x]?x:pa[x]=find(pa[x]);//路径压缩,不断的将上一级改成根节点(递归)
}int main()
{int n,k,ni,ap;while(scanf("%d",&n)==1&&n){k=0;scanf("%d%d",&ni,&ap);for(int i=1; i<=n; i++)scanf("%d%d",&dist1[i],&dist2[i]);for(int i=1; i<=n; i++){for(int j=1; j<=n; j++){a[k++].xx=i;k--;a[k++].yy=j;k--;a[k++].w=sqrt((dist1[i]-dist1[j])*(dist1[i]-dist1[j])+(dist2[i]-dist2[j])*(dist2[i]-dist2[j]));}}sort(a,a+k,cmp);double cnt=sqrt((dist1[ap]-dist1[ni])*(dist1[ap]-dist1[ni])+(dist2[ap]-dist2[ni])*(dist2[ap]-dist2[ni]));for(int i=1; i<=n; i++)pa[i]=i;pa[ni]=ap;int xx=2,yy=0;while(xx<n&&yy<k){int x,y;x=find(a[yy].xx);y=find(a[yy].yy);if(x!=y) //不连通{cnt+=a[yy].w;//并查集取代标记数组用来判断两个节点是否是统一子集pa[x]=y; //将连通分支合并}yy++; //继续下一个}printf("%.2lf\n",cnt);}return 0;
}
这篇关于HDU 4463--最小生成树+并查集标记的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!