朴素版prim算法求最小生成树

最新推荐文章于 2024-04-13 14:59:05 发布

剑决浮云气

最新推荐文章于 2024-04-13 14:59:05 发布

阅读量67

点赞数 1

分类专栏：数据结构与算法文章标签：算法图论数据结构

本文链接：https://blog.csdn.net/ZZHinclude/article/details/112739150

版权

数据结构与算法专栏收录该内容

40 篇文章 2 订阅

订阅专栏

无向图适用于稠密图时间复杂度O(N^2)

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

const int INF = 0x3f3f3f3f;
const int N = 510;

bool IsVisit[N];//判断节点是否已经加入集合 
int Graph[N][N];//邻接矩阵存储图 
int n,m;//n为节点数 m为边数 
int Dis[N];//存储节点到当前集合的最短距离 
 
int Prim()
{
	memset(Dis, INF, sizeof(Dis));
	
	//记录最小生成树的路径长度 
	int nResult = 0;
	//n次迭代 每次加入一个点 直到把所有点都加入最小生成树 
	for(int i = 0; i < n; i++)
	{
		int nShort = -1;
		
		//遍历找到当前距离集合最近的点 
		for(int j = 1; j <= n; j++)
		{
			//没有加入到集合 并且 最短的那个 
			if(!IsVisit[j] && (-1 == nShort || Dis[nShort] > Dis[j]))
			{
				nShort = j;
			} 
		}
		
		IsVisit[nShort] = true;
		
		//没有最小生成树 存在不连通 
		if(i && (INF == Dis[nShort]))
		{
			return INF;
		}
		
		//第一次不加 
		if(i)
		{
			nResult += Dis[nShort];
		}
		 
		
		//用新加入集合的节点 更新其余节点到集合的距离 
		//注意此处与Dijkstra算法的区别 
		//若存在自环 当j和nShort相等时 负环一定会影响Dis[j] 
		//所以要先加Dis[nShort] 
		for(int j = 1; j <= n; j++)
		{
			Dis[j] = min(Dis[j],Graph[nShort][j]);
		}
		
	}
	return nResult;
}


int main(int argc, char** argv)
{ 
	scanf("%d%d",&n, &m);
	
	memset(Graph, INF, sizeof(Graph));
	
	for(int i = 0; i < m; i++)
	{
		int nStart,nEnd,nWeight;
		scanf("%d%d%d",&nStart, &nEnd, &nWeight);
		//无向图 且存在重边 
		Graph[nStart][nEnd] = Graph[nEnd][nStart] =
		 min(Graph[nStart][nEnd], nWeight);
	}
	
	int nResult  = Prim();
	if(INF == nResult)
	{
		printf("No Tree\n");
	}
	else
	{
		printf("%d",nResult);
	}
	
	return 0;
}