最小生成树之Prim算法C++实现

Prim算法的基本思路：

将图中的所有的顶点分为两类：树顶点（已经被选入生成树的顶点）和非树顶点（还未被选入生成树的顶点）。首先选择任意一个顶点加入生成树，接下来要找出一条边添加到生成树，这需要枚举每一个树顶点到每一个非树顶点所有的边，然后找到最短边加入到生成树。依次，重复操作n-1次，直到将所有顶点都加入生成树中。

#include<iostream>
using namespace  std;
int main()
{
	int points,edges,min;
	int edge[7][7];
	int dis[7];
	int flag[7]={0};
	int infinity=99999999;
	cin>>points>>edges;
	for (int i=1;i<=points;i++)
	{
		for (int j=1;j<=points;j++)
		{
			if (i==j)
			{
				edge[i][j]=0;
			}
			else
			{
				edge[i][j]=infinity;
			}
		}
	}
	int p1,p2,w;
	for (i=1;i<=edges;i++)
	{
		cin>>p1>>p2>>w;
		edge[p1][p2]=w;
		edge[p2][p1]=w;
	}
	for (i=1;i<=points;i++)//初始化1号顶点到其它顶点的距离
	{
		dis[i]=edge[1][i];
	}
	flag[1]=1;
	int count=1;
	int j;
	int sum=0;
	while (count<points)
	{
		min=infinity;
		for (i=1;i<=points;i++)
		{
			if (flag[i]==0&&dis[i]<min)//每次选择1号顶点到其它顶点的最短距离加入生成树
			{
				min=dis[i];
				j=i;
			}
		}
		flag[j]=1;
		count++;
		sum+=dis[j];
		for (i=1;i<=points;i++)//选择出的顶点再延伸更新1号顶点到其它顶点的距离
		{
			if (flag[i]==0&&dis[i]>edge[j][i])
			{
				dis[i]=edge[j][i];//如果满足条件则更新
			}
		}
	}
	cout<<sum<<endl;
}

输出结果