Skip to content

Latest commit

 

History

History
64 lines (56 loc) · 1.33 KB

prim算法.md

File metadata and controls

64 lines (56 loc) · 1.33 KB

#prim算法

注意最后输出答案,要先把prim的值赋给一个变量,再输出那个变量 然后要注意过程中有无使用两等号,如果有注意是否满足bool要求 prim算法的原理是,构造最小生成树的点集合,当点集合为满时结束过程。每一次选择到点集合距离最短的点进入,并将对应的边加上,生成新的树。每选择一次节点,需要更新一次每个点到集合的距离。

模板:

# include<bits/stdc++.h>
# include<algorithm>
typedef long long ll;
int n,m;
int p[5100][5100];
int dis[5100];
bool istrue[5100];
using namespace std;
int prim()
{
	int res=0;
	memset(dis,0x3f,sizeof dis);      //在此注意要将dis距离初始化为无穷大
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!istrue[j]&&(t==-1||dis[j]<dis[t]))
			t=j;
		}
		if(i&&dis[t]==0x3f3f3f3f)     //第一次的时候的距离必然为无限大,所以需要特判
		return 114514;
		if(i)
		res+=dis[t];
		istrue[t]=true;
		for(int j=1;j<=n;j++)
		dis[j]=min(dis[j],p[t][j]);
		
	}
	return res;
}

int main (void)
{
	cin>>n>>m;
	memset(p,0x3f,sizeof p);
	
	for(int i=0;i<m;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		p[a][b]=p[b][a]=min(p[a][b],c);
	}
	int t=prim();
	if(t==114514)
	{
		cout<<"orz"<<endl;
		return 0;
	}
	cout<<t<<endl;
	return 0;
	
}