最小生成树-Prim算法

算法概述

Prim算法总是维护最小生成树的一部分。

时间复杂度O（n^2），可以二叉堆优化到O（mlogn）[但是不如直接kruskal]

所以主要用于稠密图，尤其是完全图的最小生成树求解。

算法思路

最初，Prim算法仅确定1号节点属于最小生成树。

在任意时刻，设已经确定属于最小生成树的节点集合为T，剩余节点集合为S。

找到两个端点分别属于S,T的权值最小的边，然后把属于S的端点加入T，并把z累加到答案中。

具体来说可以维护数组d：1.若x∈T，d[x]表示x被加入T时选出的最小边的权值

2.若x∈S，d[x]表示节点x和集合T中的节点之间权值最小的边的权值

代码实现

//by ziwan Catherine
//Prim算法 求稠密图的最小生成树
//在任意时刻，设已经确定属于最小生成树的节点集合为T，剩余节点集合为S
//找到两个端点分别属于S,T的权值最小的边，然后把属于S的端点加入T，并把z累加到答案中。// 
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define MAX_N 3010
using namespace std;
int a[MAX_N][MAX_N],d[MAX_N],n,m,ans;
//.若x∈T，d[x]表示x被加入T时选出的最小边的权值
//若x∈S，d[x]表示节点x和集合T中的节点之间权值最小的边的权值
bool v[MAX_N];
void prim(){
	memset(d,0x3f,sizeof(d));
	memset(v,0,sizeof(v));
	d[1]=0;
	for(int i=1;i<n;i++){
		int x=0;
		for(int j=1;j<=n;j++)
		if(!v[j]&&(x==0||d[j]<d[x])) x=j;
		v[x]=1;
		for(int y=1;y<=n;y++)
		if(!v[y]) 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;
		scanf("%d%d%d",&x,&y,&z);
		a[y][x]=a[x][y]=min(a[x][y],z);
	}//构建邻接矩阵
	prim();
	for(int i=2;i<=n;i++) ans+=d[i];
	cout<<ans<<endl; 
	return 0;
}