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

来源：互联网发布：淘宝开店流程视频编辑：程序博客网时间：2024/04/27 20:01

这是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;}