poj1258--数据结构prim算法

最新推荐文章于 2025-09-07 20:32:33 发布

zzc2227372

最新推荐文章于 2025-09-07 20:32:33 发布

阅读量150

点赞数

文章标签：数据结构与算法

本文提供了一个使用C++实现的Prim算法模板代码，用于寻找加权无向图中的最小生成树。通过不断选择最小权重的边并将顶点加入到生成树中，直至包含所有顶点。

此题就是prim算法模板

直接给代码、

#include "iostream"
using namespace  std;
#define MAXSIZE 102
int map[MAXSIZE][MAXSIZE];
int vis[MAXSIZE];
int temp[MAXSIZE];
int main()
{
	int n;
	while (scanf("%d",&n)!=EOF)
	{
		int  ans=0;
		int i,j;
		for (i=0;i<n;i++)
		{
			for (j=0;j<n;j++)
			{
				scanf("%d",&map[i][j]);
			}
		}

		memset(vis,0,sizeof(vis));
		for (j=0;j<n;j++)
		{
			temp[j]=9000010;
		}
		i=0;
		for (j=0;j<n;j++)
		{
			if (map[i][j]!=0)
				if (map[i][j]<temp[j])
				{
					temp[j]=map[i][j];
				}
		}
		vis[0]=1;
		while (i<n-1)
		{
			
			//找最小的边
			int min=9000010;
			int minindex=0;
			for (j=0;j<n;j++)
			{
				if (vis[j]==1)
				{
					continue;
				}
				if (min>temp[j])
				{
					min=temp[j];
					minindex=j;
				}
			}
			
			
			//把最小的边的顶点加入集合 
			vis[minindex]=1;
			
			//更新temp
			for (j=0;j<n;j++)
			{
				if (map[minindex][j]!=0&&vis[j]!=1)
					if (map[minindex][j]<temp[j])
					{
						temp[j]=map[minindex][j];
					}
			}
			
			
			ans+=min;
			i++;
			
			
		}
		printf("%d\n",ans);

	}
	
	return 0;	
}

/*
6 
0 6 1 5 0 0
6 0 5 0 3 0
1 5 0 5 6 4
5 0 5 0 0 2
0 3 6 0 0 6
0 0 4 2 6 0
*/