Prim

Prim模板

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

// T:确定属于最小生成树的节点集合
// S:剩余节点集合 
int a[3010][3010];
int d[3010];//  d[x], x表示x加入T时选出的最小边的权值 
bool vis[3010];// vis[x]==true表示该节点已经加入了T集合 
int n,m,ans=0;

void prim()
{
	memset(d,0x3f,sizeof(d));
	memset(vis,0,sizeof(vis));
	d[1]=0;
	for(int i=1; i<n; i++)//循环n-1次  每次找出一个点加入T 
	{
		int x=0;
		for(int j=1; j<=n; j++)//遍历所有节点 
		{
		  //当前d[]数组存储的是各个节点距离生成树的最小权值 
			if(!vis[j] && (x==0 || d[j]<d[x])) //!vis[j]表示该节点尚未加入T集合 
				x=j;                         //选出距离生成树最近的节点 
		}
		vis[x]=1;//将该节点加入到生成树中(T集合)  
		for(int y=1; y<=n; y++)
		{
			if(!vis[y])//新加了一个点，就维护该点到还没有加入T集合的点的距离  
				d[y]=min(d[y],a[x][y]);//取距离生成树 
		}
	}
}
int main()
{
	cin>>n>>m;
	//构建邻接矩阵
	memset(a,0x3f,sizeof(a));
	for(int i=1; i<=n; i++) a[i][i]=0;
	for(int i=1; i<=m; i++)
	{
		int x,y,z;
		cin>>x>>y>>z;
		a[x][y]=a[y][x]=min(a[x][y],z);
	}

	//求最小生成树
	prim();
	for(int i=2; i<=n; i++) ans+=d[i];
	cout<<ans<<endl;
	return 0;
}

/*  输入  输出15
6 10
1 2 6
1 3 1
1 4 5
2 3 5
2 5 3
3 4 5
3 5 6
3 6 4
4 6 2
5 6 6
*/