最小生成树Prim算法模板

/*

Prim:利用dis[]来记录已选点集中到各点的最短距离，一开始用起始点到各点的距离初始化dis[]

while->知道全部点都走到

for->找dis[]中最小边，添加长度，添加点进已选点集

for->利用刚添加进来的点到各点的距离，更新dis[]

*/

#include<iostream>

using namespace std;
#define N 28
int map[N][N];
int state[N];
int Prim(int maxnum)
{
int num=1;
int sum=0;
int min;
int i,j;
int dis[N];
dis[0]=0;
state[0]=1;
for(i=1;i<maxnum;i++) dis[i]=map[0][i];
while(num!=maxnum)
{
min=1000;
for(i=0;i<maxnum;i++)
{
if(state[i]==0&&dis[i]!=-1&&dis[i]<min)
{
min=dis[i];
j=i;
}
}
sum+=min;
num++;
state[j]=1;
for(i=0;i<maxnum;i++)
{
if(state[i]==0&&map[j][i]!=-1)
{
if(dis[i]==-1) dis[i]=map[j][i];
else if(map[j][i]<dis[i]) dis[i]=map[j][i];
}
}
}
return sum;
}
int main()
{
int n,m,l,i,j;
char ch,ch1;
while(cin>>n)
{
if(n==0) break;
memset(map,-1,sizeof(map));
memset(state,0,sizeof(state));
for(j=0;j<n-1;j++)
{
cin>>ch>>m;
for(i=0;i<m;i++)
{
cin>>ch1>>l;
map[int(ch-'A')][int(ch1-'A')]=l;
map[int(ch1-'A')][int(ch-'A')]=l;
}
}
cout<<Prim(n)<<endl;
}
return 0;
}