prim（详细代码）+HDU1233还是畅通工程

以前的时候可不理解prim代码怎么写。。。但是原理懂。。。
现在看来不会写代码说明原理懂得还是不够透彻。。。。直接压根不懂吧…

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5

/*prime的思路 有两个集合假设P(vis)为定点集合，Q为未确定点的集合 下面程序是把定点标记为1 （初始距离+n-1循环+选择到定点到Q中最短距离+更新P与Q之间的距离）
（1）先选一个点（任意）为定点
（2）记录这个点到Q中其他点的距离（dis)
(3) 行n-1次循环，加入剩下的n-1个点
(4)选择该定点到Q中其他点距离就近的点 （即dis中最小的）
（5）更新P中定点到Q中点的距离 （与（2）差不多）
*/
#include<stdio.h>
#include<iostream>
#include<string.h>
#define INF 0x7fffffff
using namespace std;

int map[110][110],dis[110],vis[110],n;// map矩阵便于存放各个点之间的距离 n全局变量
//void prim()
int prim(int n){
memset(vis,0,sizeof(vis));//初始化vis数组
vis[1]=1;//1开始（可以任意）标记为1
for(int i=1;i<=n;i++){
dis[i]=map[1][i];//这个时候记录的是定点1到其他点的距离（也就是矩阵第1行的所有元素)
}
int sum=0;
for(int k=1;k<=n-1;k++){//进行n-1次循环，加入剩下的n-1个点~
int minn=INF,t;
for(int i=1;i<=n;i++){
if(!vis[i]&&dis[i]<minn){//选择到vis里点的最近距离
minn=dis[i];
t=i;//此时记录那个点与定点的距离最近，以便于标记
}
}
sum+=minn;//求和 循环 n次得到的最近距离
vis[t]=1; //标记
for(int i=1;i<=n;i++){//更新定点vis和未确定点之间的距离 如何更新的呢？刚开始我也不太了理解，后来恍然大悟~
if(!vis[i]&&dis[i]>map[t][i]){/*比如 刚刚咱们不是已经得到一组数组dis存放的是定点1到其他点的距离吗，假设这个时候又来了一个定点2，
那么需要比较定点1,2到其他点的最短距离 一般的话 我们这个时候会去取矩阵的第一行和第二行（二维） 然后一个一个去比较谁小
然而这里取得还是一维数组 咋样去把二维改成一维的呢？ 比如 1到3的距离是12,2到3的距离是10，那么这个时候我们只需要取2到3的距离10就行了 依次类推*/
dis[i]=map[t][i];
}
}
}
return sum;
//cout<<sum<<endl;
}

int main(){
//	int N;全局变量
while(~scanf("%d",&n)&&n){
for(int i=1;i<=n;i++){//初始化矩阵map
for(int j=1;j<=n;j++){
map[i][j]=INF;
}
}
int a,b,dis;
for(int i=0;i<n*(n-1)/2;i++){
scanf("%d %d %d",&a,&b,&dis);
map[a][b]=map[b][a]=dis;//比如1到3的距离是250，那么矩阵中第1行第3列对应的位置是250 相应的3行第1列应的位置也是250 两者相等都是250
}
printf("%d\n",prim(n));
//		  prim();
}
return 0;
}

• 点赞
• 收藏
• 分享
• 文章举报