最小生成树-Prim算法模板

最小生成树-Prim算法：
未优化：时间复杂度 O(n^2),n为顶点数
堆优化：时间复杂度大概为 O((n+m)logm),m为边数
模板题：P3366 【模板】最小生成树

Code 未优化:

#include<iostream>
#include<vector>
using namespace std;
typedef pair<int,int> pr; 

const int MAX_N=5e3+5;
const int INF=1e9+5;
int n,m;
vector<pr> e[MAX_N];
int dist[MAX_N];
bool tag[MAX_N];

int Prim(){
	int res=0;
	for(int i=1;i<=n;++i)
		dist[i]=INF;
	dist[1]=0;
	int u,Min;
	for(int i=0;i<n;++i)
	{
		u=-1; Min=INF;
		for(int j=1;j<=n;++j)
			if(!tag[j]&&dist[j]<Min){
				Min=dist[j];	u=j;
			}
		if(u==-1){
			res=-1;	break;
		}
		res+=Min;	tag[u]=true;
		for(int j=0;j<e[u].size();++j)
			dist[e[u][j].first]=min(dist[e[u][j].first],e[u][j].second);
	}
	return res;
}

int main()
{
	scanf("%d%d",&n,&m);
	int u,v,w;
	for(int i=0;i<m;++i)
	{
		scanf("%d%d%d",&u,&v,&w);
		e[u].push_back({v,w});
		e[v].push_back({u,w});
	}
	int res=Prim();
	if(res!=-1)	printf("%d\n",res);
	else printf("orz\n");
	
	return 0;
}

Code 堆优化：

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
typedef pair<int,int> pr;

const int MAX_N=1e5+5;
const int INF=1e9+5;
int n,m;
vector<pr> e[MAX_N];
int dist[MAX_N];
bool tag[MAX_N];
priority_queue<pr,vector<pr>,greater<pr>> Q;

int Prim(){
	int res=0;
	for(int i=1;i<=n;++i)
		dist[i]=INF;
	dist[1]=0;
	Q.push({0,1});
	int u,v,w,cnt=n;
	while(!Q.empty()&&cnt){
		u=Q.top().second;	w=Q.top().first;	Q.pop();
		if(tag[u])	continue;
		res+=w;	tag[u]=true;	--cnt;
		for(int j=0;j<e[u].size();++j)
		{
			v=e[u][j].first;	w=e[u][j].second;
			if(!tag[v]&&dist[v]>w){
				dist[v]=w;
				Q.push({dist[v],v});
			}
		}
	}
	if(cnt)	res=-1;
	return res;
}

int main()
{
	scanf("%d%d",&n,&m);
	int u,v,w;
	for(int i=0;i<m;++i)
	{
		scanf("%d%d%d",&u,&v,&w);
		e[u].push_back({v,w});
		e[v].push_back({u,w});
	}
	int res=Prim();
	if(res!=-1)	printf("%d\n",res);
	else printf("orz\n");
	
	return 0;
}