最小生成树 Prim算法

最新推荐文章于 2022-05-09 22:58:13 发布

_北鸣

最新推荐文章于 2022-05-09 22:58:13 发布

阅读量591

点赞数

分类专栏： OJ训练文章标签：最小生成树 Prim

本文链接：https://blog.csdn.net/MeJustOny/article/details/53240329

版权

OJ训练专栏收录该内容

78 篇文章 0 订阅

订阅专栏

/***************
Problem from :ds
Problem describe :最小生成树 
****************/
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#include<stack>
#include<queue>
#include<ctime>
#include<cstring>
#include<string>
#define LL __int64
#define INF 0x3f3f3f3f3f
#define maxn 5000+5
using namespace std;
int edge[maxn][maxn]={0};
int used[maxn]={0}, dis[maxn]={0};//used标记该点是否在最小生成树集合中, dis[i]表示编号为i的点到最小生成树集合的最短距离
int main()
{
// 	freopen("in.txt","r",stdin);
//      freopen("out.txt","w",stdout);
	int i, j, k, n, m, u, v, w, Min=INF;
	memset(dis, INF, sizeof(dis));
	memset(edge, INF, sizeof(edge));
	scanf("%d%d", &n, &m);
	for(i=0; i<m; i++)
	{
		scanf("%d%d%d", &u, &v, &w);
		edge[u][v] = edge[v][u] = w;
		if( w < Min ) j=u, k=v, Min=w;//找到最小权值的边
	}
	int len=0, ans=0;
	used[j] = used[k] = 1;//将最小权值边的两个顶点标记为已经在最小生成树中
	ans += Min;
	for(i=1; i<=n; i++)//此循环为了将除了已经在最小生成树中外其他顶点到最小生成树集合的最短距离dis
	{
		if(used[i]) continue;
		if(edge[i][j] < dis[i] ) dis[i] = edge[i][j];
		if(edge[i][k] < dis[i] ) dis[i] = edge[i][k];
	}
	for(i=1; i<=n; i++)
	{
		Min = INF;
		int pos=-1;
		for(j=1; j<=n; j++)
		{
			if(!used[j] && dis[j] < Min) //找到距离最小生成树集合最近的顶点/该顶点不能已经在最小生成树的
			{
				Min = dis[j];
				pos = j;
			}
		}
		if(pos==-1) break;//所有顶点都进入最小生成树集合了,所以pos没有变化,还是-1
		used[pos] = 1;
		ans += dis[pos];
		for(j=1; j<=n; j++)//更新其他顶点到最小生成树集合的dis
		{
			if(!used[j] && dis[j]>edge[j][pos])
			{
				dis[j] = edge[j][pos];
			}
		}
	}
	printf("%d\n", ans); 
}