【算法练习】最小生成树

题意：【模板】最小生成树

方法1：Prim算法(稠密边用优）

#include <bits/stdc++.h>
using namespace std;
int n,m,u,v,d,ans;
bool f[5001];
vector<pair<int,int>> a[5001];//用结构体和重载比直接定义小根堆似乎还快一点点
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> q;//直接定义小根堆优选队列
int main(){
	scanf("%d%d",&n,&m);
	while(m--){
		scanf("%d%d%d",&u,&v,&d);
		a[u].push_back({v,d});//双向存点和权值
		a[v].push_back({u,d});
	}
	f[1]=1;//1号节点赋初值，表示已访问
	//初始化将1号节点对应的边放入队列，权值放前便于优先队列按权值从小到大排序
	for(auto &v:a[1]) q.push({v.second,v.first});
	for(int i=1;i<n;i++){
		while(!q.empty()&&f[q.top().second]) q.pop();
		if(q.empty()){
			printf("orz");
			return 0;
		}
		v=q.top().second,d=q.top().first,q.pop();
		ans+=d,f[v]=1;//找到一个点，增加权值、标已访问
		for(auto &u:a[v]) q.push({u.second,u.first});//将当前个节点连通的点入队
	}
	printf("%d",ans);
	return 0;
}

方法2：Kruskal算法(稀疏边用优）

#include <bits/stdc++.h>
using namespace std;
int n,m,u,v,d,cnt,ans,f[5001];
struct node{
	int x,y,dis;
	bool operator < (const node& i) const{
		return dis>i.dis;
	}
};
priority_queue<node> q;//未加参数，默认为大根堆优先队列
int myfind(int x){ 
	return x==f[x]?x:f[x]=myfind(f[x]);
}
void Kruskal(){
	for(int i=1;i<=n;i++) f[i]=i;//初始化并查集
	while(!q.empty()&&cnt<n-1){
		int fx=myfind(q.top().x),fy=myfind(q.top().y),d=q.top().dis;
		q.pop();
		if(fx!=fy)
			f[fy]=fx,ans+=d,cnt++;
	}
}
int main(){
	scanf("%d%d",&n,&m);
	while(m--){
		scanf("%d%d%d",&u,&v,&d);
		q.push({u,v,d});//直接存边和权值，与prim算法区别
	}
	Kruskal();
	if(cnt==n-1) printf("%d",ans);
	else printf("orz");
	return 0;
}

谢谢！