最小生成树prim算法模板及详解

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 2020;
int g[N][N];
int dist[N];
bool st[N];
int len = 0;
int n,m;
int prim()
{
	dist[1] = 0;
	for(int i=0;i<n;i++)
	{
		int t = -1;
		for(int j=1;j<=n;j++)
		if(!st[j] && (t==-1 || dist[t]>dist[j]))
		t = j;
		if(i && dist[t]==0x3f3f3f3f) return -1; //这句是用来判断是不是联通的，以便于及时退出
		len += dist[t];
		st[t] = 1;
		for(int j=1;j<=n;j++)                //每次选出一个点后都更新一下周围的邻点，相当于把点集s看成是一个集合 
		dist[j] = min(dist[j],g[t][j]);      //集合中的任意点都可以用来更新其周围的点，所以最后的结果就是集合中所有点能更新的都更新了之后，这点离整个集合的最小值 
	}
	return len;
} 
int main()
{
	memset(dist,0x3f3f3f3f,sizeof dist);
	memset(g,0x3f3f3f3f,sizeof g);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		g[a][b] = g[b][a] = min(g[a][b],c);
	}
	int t = prim();
	if(t==-1) cout<<"impossible";
	else cout<<t;
	return 0;
} 

prim算法和朴素dijkstra算法及其相近

思想是这样的：prim也是基于贪心算法，用更新过的点的集合（姑且叫做s集合）来更新集合外的点

但是dijkstra算法是搜集合外的点到起点的最小距离，prim算法是搜集合外的点到集合的最小距离，但是这个集合怎么表示呢？

就是用集合内的点都更新其邻点就好，集合内的点可能有多个都可以到达a点，那么就保存其多次更新后的最小值，以此当做a点到集合的最小距离

但是还是要多写一句话方便及时退出：其一是若是集合内的点的距离是正无穷，那么必然不可能实现最小生成树