prim算法最小生成树

# include<iostream>
# include<cstring>
# define inf 0x3f3f3f3f
using namespace std;
const int maxn=100+5;
typedef struct node{
	int adj; //顶点之间的权重 
}node;
typedef struct graph{
	int vertex[maxn];
   node vertexs[maxn][maxn];//邻接矩阵 
    int vertexnum;
    int edgenum;
}graph;
struct closeedge {
	int adjvex;
	int lowcost;
};
closeedge ss[maxn]; 
int find(graph g,int i)  //寻找对应顶点的索引 
{
	for(int j=0;j<g.vertexnum;j++)
	    if(g.vertex[j]==i)
	      return j;
	return -1;
}
void init(graph &g)   //初始化图 
{
	cin>>g.vertexnum>>g.edgenum;
	for(int i=0;i<g.vertexnum;i++)
	  cin>>g.vertex[i];
	for(int i=0;i<g.vertexnum;i++)
	  for(int j=0;j<g.vertexnum;j++)
	      g.vertexs[i][j].adj=inf;
	for(int i=0;i<g.edgenum;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
	    int index_u=find(g,u);
		int index_v=find(g,v);
		if(index_u==-1||index_v==-1)
		  {
		  	  cout<<"不存在该节点"<<endl;
		  	 continue;
			} 
		g.vertexs[index_u][index_v].adj=w;
		g.vertexs[index_v][index_u].adj=w;   	
	  }
}
int minindex(int x,closeedge *ss)  //找到最小边的索引 
{
	int m_in=inf;
	int index=-1;
	for(int i=0;i<x;i++)
	{
		if(ss[i].lowcost>0&&m_in>ss[i].lowcost)     
		 {
		 	 m_in=ss[i].lowcost;
		 	 index=i;
		   } 
		   
	}
	return index;
}
void prim(graph g,int w,closeedge *ss)
{
	int k=find(g,w);
	for(int i=0;i<g.vertexnum;i++)  //初始化，求得其他顶点到原顶点的距离。 
	  if(i!=k)
	  {
	  	ss[i].adjvex=w;
	  	ss[i].lowcost=g.vertexs[k][i].adj;
	  }
	ss[k].lowcost=0;    //作用：（1）加入集合u中.（2）作为已加入集合中的标志 
	for(int i=1;i<g.vertexnum;i++)   //选择除去原顶点以外的其他顶点 
	{
		int ww=minindex(g.vertexnum,ss);    //选择最小边的索引 
		cout<<"(V"<<ss[ww].adjvex<<",V"<<g.vertex[ww]<<") ";
		ss[ww].lowcost=0;//加入集合u中
		//以下代码很关键。 
		for(int j=0;j<g.vertexnum;j++)
		{
			if(g.vertexs[ww][j].adj<ss[j].lowcost)  //如果此时的ww到j的距离比上一个到j的距离还要小，就做如下更新操作 
			{
				ss[j].adjvex=g.vertex[ww];
				ss[j].lowcost=g.vertexs[ww][j].adj;
			}
		}
	}
}
int main()
{
	graph g;
	int w;
	init(g);
	cin>>w;
	prim(g,w,ss);
	return 0;
}