prim算法来解最小生成树问题

#include<iostream>
using namespace std;
int main()
{
	int graph[100][100],dis[100],book[100]={0};//dis数组是记录生成树到非树顶点的距离，book记录生成树的顶点
	int inf=999999;
	int count=0,sum=0;//count用来记录生成树中顶点的个数，sum用来存储路径之和
	int n,m;  //n表示图的顶点个数，m表示图的边的个数
	cin>>n>>m;
	
	for(int i=1;i<=n;i++) //初始化图的邻接矩阵 
	for(int j=1;j<=n;j++)
	if(i==j) graph[i][j]=0;
	else graph[i][j]=inf;
	
	for(int i=1;i<=m;i++) //开始读入边 
	{
		int t1,t2,t3;
		cin>>t1>>t2>>t3;
		graph[t1][t2]=t3;
		graph[t2][t1]=t3;
	 }
	 
	 for(int i=1;i<=n;i++) //初始化dis数组，这里是1号顶点到各个顶点的初始距离，因为当前生成树中只有1号顶点 
	 dis[i]=graph[1][i];
	 
	 //prim算法的核心部分
	 book[1]=1;      //这里用book来标记一个顶点是否已经加入生成树
	 count++;
	 
	 while(count<n)
	 {
	 	int min=inf,j;
	 	for(int i=1;i<=n;i++)
	 	{
	 		if(book[i]==0&&dis[i]<min)
	 		{
	 			min=dis[i];
	 			j=i;
			 }
		 }
		 
		 book[j]=1;
		 count++;
		 sum+=dis[j];
		 
		 for(int k=1;k<=n;k++)  //扫描当前顶点j所有的边，再以j为中间点，更新生成树到每一个非树顶点的距离
		 {
		 	if(book[k]==0&&dis[k]>graph[j][k])
		 	dis[k]=graph[j][k];
		  } 
	  }
	  cout<<sum;
	  return 0; 
}