[ACM]TL-Prim

最新推荐文章于 2025-11-26 15:40:52 发布

原创最新推荐文章于 2025-11-26 15:40:52 发布 · 107 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍了一个使用C++实现的Prim算法，该算法用于寻找加权图中的最小生成树。通过输入顶点数和边数，初始化图的邻接矩阵，并通过Prim算法逐步构建最小生成树，最终输出最小生成树的总权重。

#include<iostream>
#include<cstdio>

using namespace std;

int main(){
	int inf = 99999999;
	int n,m,t1,t2,t3,min;
	int e[7][7],dis[7],book[7]={0};
	int count=0,sum=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(i==j)	e[i][j]=0;
			else e[i][j]=inf;
		}
	}
	int j;
	for(int i=1;i<=m;i++)
	{
		cin>>t1>>t2>>t3;
		e[t1][t2]=t3;
		e[t2][t1]=t3;
	}
	for(int i=1;i<=n;i++)
	{
		dis[i]=e[1][i];
	}
	
	
	//Prim
	book[1]=1;
	count++;
	while(count<n)
	{
		min=inf;
		for(int i=1;i<=n;i++)
		{
			if(book[i]==0 && dis[i] < min)
			{
				min=dis[i];
				j=i;
			}
		}
		book[j]=1;
		count++;
		sum+=dis[j];
		
		for(int k=1;k<=n;k++)
		{
			if(book[k]==0 && dis[k]>e[j][k])
				dis[k]=e[j][k];
		}
	}
	
	cout<<sum;
	return 0;
}