普里姆(prim)算法是一种用来查找最小生成树的算法。
流程如下:
1.先存图,存边(用邻接矩阵将出入点和边权存起来)。
2.然后选定一个起点加入边集(比如从1开始)。
3.然后以边集里面的为起点寻找最小的边权,然后将终点加入到边集里。
4.重复步骤三,直到找到最小生成树。
5.最后将所有边集加起来。
如下图所示:
prim算法

#include<bits/stdc++.h>
using namespace std;
int a[3001][3001],d[3010],n,m,ans;
bool v[3010];
void prime(){
	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(){
	scanf("%d%d",&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];
	}
	prime();
	for(int i=2;i<=n;i++) ans+=d[i];
	cout<<ans<<endl;
}

相关文章: