hdu1875浅谈prim算法的朴素实现
阅读原题
题目大意
给你几个（<=100）小岛的坐标，然后你把所有的岛都修上桥连接起来，求最小花费，还有个附加的限制：只有岛之间的距离大于等于10，或小于等于1000时才能修桥。
大概是因为十米以内不用建桥，千米以上无法建桥。哈哈，说着玩的。
很明显这是一道MST（最小生成树）的题目，貌似也有人用并查集AC过。
最小生成树算法
概述
最小生成树的常用算法有两个kruskal和prim算法。两者都是不停地执行归并操作，然而一言以蔽之，两者的不同之处在于：kruskal----归并边；prim----归并点
关于kruskal的题目原先AC过几道：
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不了的。大家具体方案具体分析啦。
