Prim_邻接矩阵实现

#include<iostream>
#include<limits.h>
using namespace std;
const int maxn=101;
int n,sum=0,dist[maxn],map[maxn][maxn]={INT_MAX},pre[maxn];
int Prim(int n,int dist[maxn],int map[maxn][maxn],int pre[maxn])
{
int i,j,k,min;
bool p[maxn];
for(i=2;i<=n;i++)
{
p[i]=false;
dist[i]=map[1][i];
pre[i]=1;
}
p[1]=true;
dist[1]=0;
for(i=1;i<n;i++)
{
min=INT_MAX;
k=0;
for(j=1;j<=n;j++)
{
if(!p[j]&&dist[j]<min)
{
min=dist[j];
k=j;
}
}
if(k==0) return -1;
p[k]=true;
for(j=1;j<=n;j++)
{
if(!p[j]&&map[k][j]!=INT_MAX&&map[k][j]<dist[j])
{
dist[j]=map[k][j];
pre[j]=k;
}
}
}
int sum=0;
for(int p=1;p<=n;p++)
sum+=dist[p];
return sum;
}
int main()
{
int m;
cin>>n>>m;
for(int i=0;i<m;i++)
{
int a,b,c;
cin>>a>>b>>c;
map[a][b]=map[b][a]=c;
}
for(int i=0;i<n;i++)
map[i][i]=INT_MAX;
int sum=Prim(n,dist,map,pre);
cout<<sum;
}