Prim算法求最小生成树

---

一、Prim算法求最小生成树

1).输入：一个加权连通图，其中顶点集合为V，边集合为E；
2).初始化：Vnew = {x}，其中x为集合V中的任一节点（起始点），Enew = {},为空；
3).重复下列操作，直到Vnew = V：
a.在集合E中选取权值最小的边<u, v>，其中u为集合Vnew中的元素，而v不在Vnew集合当中，并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
b.将v加入集合Vnew中，将<u, v>边加入集合Enew中；
4).输出：使用集合Vnew和Enew来描述所得到的最小生成树。

二、代码实现

#include<bits/stdc++.h>
using namespace std;

//dist[i]初始化为正穷
//不在集合中的最小的点，赋值给t;
//把t加到集合当中去。
//用t来更新其他点到集合的距离。
const int N=0x3f3f3f;
const int k=200005;
int dist[5005];
bool b[50005];
int g[5005][5005];
int n,m;
//prim算法
//先初始距离数组为正无穷
//定义结果，r记录返回结果
//进行遍历
//{
//	先声明一个变量t用来记录点的值,t=-1 
//	
//} 

int prim()
{
	memset(dist,N,sizeof dist);
	int res=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++) 
		{
			if(!b[j]&&(t==-1||dist[t]>dist[j]))
				t=j;
		}
		if(i&&dist[t]==N) return N;
		if(i) res+=dist[t];
		b[t]=true;
		for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
	}
	return res;
}



int main()
{
	//先输入 n个点，m条边
//	将邻接矩阵初始化，每条边都是最大值，memset(g,0x3f,sizeof g)
//	输入m条边，注意有重边
//	进行prim算法
//	如果返回的是一个无穷大，则说明不是连通图
//	不是的话，就输出值。 
	cin>>n>>m;
	memset(g,N,sizeof g);
	for(int i=0;i<m;i++)
	{
		int x,y,c;
		cin>>x>>y>>c;
		g[x][y]=g[y][x]=min(g[x][y],c);
	}
	int t=prim();
	if(t>=N)
	{
		cout<<"orz"<<endl;
	}
	else
	{
		cout<<t<<endl;
	}
	return 0;
}

---