首页 > 代码库 > hdu1875浅谈prim算法的朴素实现
hdu1875浅谈prim算法的朴素实现
阅读原题
题目大意
给你几个(<=100)小岛的坐标,然后你把所有的岛都修上桥连接起来,求最小花费,还有个附加的限制:只有岛之间的距离大于等于10,或小于等于1000时才能修桥。
大概是因为十米以内不用建桥,千米以上无法建桥。哈哈,说着玩的。
很明显这是一道MST(最小生成树)的题目,貌似也有人用并查集AC过。
最小生成树算法
概述
最小生成树的常用算法有两个kruskal和prim算法。两者都是不停地执行归并操作,然而一言以蔽之,两者的不同之处在于:kruskal----归并边;prim----归并点
关于kruskal的题目原先AC过几道:
- hdu1301
- hdu1879
- hdu3371
prim算法
先贴一个维基百科的链接“prim算法”。下面我用离散数学来描述一下。
设有图G=(V,E),所有的结点集合为V,另有一空集合U。基本思路是:
- 先随意确定一个起点。
- 设此点为v,加入集合U中。
- 观察与U中点相连的边,找到最短边。
- 把与最短边另一端的点加入到集合U中。
- 继续执行步骤3.知道V-U=0.即所有点都加入U中。
本题AC代码
朴素实现
这是最一般的实现方案。近乎模板的实现代码,在任一本数据结构书中可能都有提到。
void prim() { double sum=0,lowcast[105]={0}; int count=1; for(int i=0;i<c;i++) { lowcast[i] = d[0][i]; } for(int i=0;i<c-1;i++) { double min = INF;//INF是自定义常量,表无穷大 int k = 105; for(int j=0;j<c;j++) if(min>lowcast[j]&&lowcast[j]) { min = lowcast[j]; k = j; } if(k!=105) { sum+=lowcast[k]; lowcast[k]=0; count++; } for(int j=0;j<c;j++) if(d[j][k]<lowcast[j]) { lowcast[j] = d[j][k]; } } if(count==c) printf("%.1lf\n",sum*100); else printf("oh!\n"); }
解释
二维数组d,保存着所有岛两两的距离。首先我们选择的起点是结点0.
for(int i=0;i<c;i++) { lowcast[i] = d[0][i]; }lowcast数组保存的是集合U中的点到V-U中每个结点的最近距离。一开始的时候,集合U中只有结点0,所以lowcast数组保存的都是结点0到每个结点的距离。
for(int i=0;i<c-1;i++) { double min = INF;//INF是自定义常量,表无穷大,实际不可能为无穷大 int k = 105;//初始化为105,因为实际不可能为105这是prim算法的主要部分开始了,初始化min用于保存到集合U中的点最小距离的边的长度。k就是用于保存最短边的另一端的结点名称。
for(int j=0;j<c;j++) if(min>lowcast[j]&&lowcast[j]) { min = lowcast[j]; k = j; }这里也比较好理解,就是在不停的更新最短边的长度及最短边的另一端结点名。注意lowcast[j]==0时,表示结点 j 已经加入集合U了。
if(k!=105) { sum+=lowcast[k]; lowcast[k]=0; count++; }如果k!=105,表示确实找到了最短边。所以把这条边的长度lowcast[k]加入到总和sum中,同时标记该点k已经加入集合U,即lowcast[k]=0。count用于记录已加入集合U中的点的个数。
for(int j=0;j<c;j++) if(d[j][k]<lowcast[j]) { lowcast[j] = d[j][k]; }因为集合U中的点可能已经增加了,所以集合U中的点到V-U的点的最近距离也要再次更新。这个操作被称为 “松弛操作”。在图论问题中有较多应用。
完整AC代码
#include <iostream> using namespace std; #include <cmath> const double INF=0x3f3f3f3f*1.0; struct node { int x; int y; }; double d[105][105]; int c;//岛屿个数 void prim() { double sum=0,lowcast[105]={0}; int count=1; for(int i=0;i<c;i++) { lowcast[i] = d[0][i]; } for(int i=0;i<c-1;i++) { double min = INF;//INF是自定义常量,表无穷大 int k = 105; for(int j=0;j<c;j++) if(min>lowcast[j]&&lowcast[j]) { min = lowcast[j]; k = j; } if(k!=105) { sum+=lowcast[k]; lowcast[k]=0; count++; } for(int j=0;j<c;j++) if(d[j][k]<lowcast[j]) { lowcast[j] = d[j][k]; } } if(count==c) printf("%.1lf\n",sum*100); else printf("oh!\n"); } int main() { int t; cin>>t; while(t--) { node n[105]; cin>>c; for(int i=0;i<c;i++) cin>>n[i].x>>n[i].y; for(int i=0;i<c;i++) for(int j=0;j<c;j++) { if(i==j) { d[i][j]=0; continue; } int x1=n[i].x, x2=n[j].x; int y1=n[i].y, y2=n[j].y; double dist=sqrt(1.0*(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)); if(dist<10.0||dist>1000.0) d[j][i]=d[i][j]=INF; else d[j][i]=d[i][j]=dist; } prim(); } return 0; }
其他注释
const double INF=0x3f3f3f3f*1.0;自定义表示无穷大的double常量。关于这个值的选取可以参考这篇文章《编程中无穷大常量的设定技巧》。
在计算岛与岛之间距离的时候。
if(i==j) { d[i][j]=0; continue; }表示如果是i==j,就直接设为0,在prim算法里对应lowcast也会是0,会被忽略掉。我这里不能设成INF,我的实现方案中INF会导致AC不了的。大家具体方案具体分析啦。
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。