[C++]第一次PRIM，写得很丑 [还不失完美，哈。]

最新推荐文章于 2023-02-15 18:34:48 发布

原创最新推荐文章于 2023-02-15 18:34:48 发布 · 612 阅读

0 ·

CC 4.0 BY-SA版权

开发里程碑同时被 2 个专栏收录

2 篇文章

订阅专栏

L'ARGENT!! -> <-

1 篇文章

订阅专栏

本文分享了一次使用PRIM算法解决最小生成树问题的经历，通过矩阵方式实现，并最终获得一遍AC的好成绩。文中详细介绍了算法的具体实现过程。

这是HDU1233的题目，第一次写PRIM，由于不熟练，对采用什么样的数据结构纠结了很久，纠结半天还是先用矩阵写出来再说。

不过还是一遍AC了，我很高兴。[网上还有用边的数组E表示的，写的很简单，我觉得很好，下次写PRIM我也要试试。]

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
int p[120][120],v[120],t[120][120]; //p原图 v标记 t生成树
int main(){
	int n,tn,a,b,ll,s,ct,mi,mj,mm,k;
	while(scanf("%d",&n)!=EOF){
		if(n==0)break;
		tn=n*(n-1)/2; ct=0;s=0; //初始化
		memset(p,-1,sizeof(p)); memset(t,0,sizeof(t));
		memset(v,0,sizeof(v)); v[0]=1;
		for(int i=0;i<tn;i++){
			scanf("%d %d %d",&a,&b,&ll); a--;b--;
			p[a][b]=p[b][a]=ll; //建图
		}
		while(ct<n-1){
			mi=-1;mj=-1;mm=100000;
			for(int i=0;i<n;i++)for(int j=0;j<n;j++)
				if(p[i,j]>=0&&v[i]+v[j]==1) //穷举与树相邻的点
					if(p[i][j]<mm){mi=i;mj=j;mm=p[i][j];} //取最短边
			k=(v[mi]?mj:mi); //printf("%d~%d\n",mi+1,mj+1);
			t[mi][mj]=p[mi][mj]; v[k]=1; ct++; //生成树
		}
		for(int i=0;i<n;i++)for(int j=0;j<n;j++)s+=t[i][j]; //求边长总和
		printf("%d\n", s);
	}
	return 0;
}