最小生成树Kruskal和Prim模板c++

P3366 【模板】最小生成树 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)https://www.luogu.com.cn/problem/P3366

prim:该算法的时间复杂度为O(n2)。与图中边数无关，该算法适合于稠密图。

kruskal:需要对图的边进行访问，所以克鲁斯卡尔算法的时间复杂度只和边又关系，可以证明其时间复杂度为O（eloge）。适合稀疏图。

kruskal ( 并查集 ) 模板:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=5001;
const int maxm=200005;
struct node{
	int u,v,w;
}a[maxm];

bool cmp(node a,node b){
	return a.w<b.w;
}

int n,m,cnt,k,ans;
int fa[maxn];

int find(int x){
	if(x==fa[x]) return x;
	fa[x]=find(fa[x]);
	return fa[x];
}

inline void init(){
	for(int i=1;i<=n;i++)
		fa[i]=i;
}

void kruskal(){
	sort(a,a+m,cmp);
	for(int i=0;i<m;i++){
		int u=a[i].u;
		int v=a[i].v;
		int w=a[i].w;
		int eu=find(u),ev=find(v);
		if(eu==ev) continue;
		ans+=w;
		fa[eu]=ev;
		if(++cnt==n-1)
			break;
	}
}

int main()
{
	cin>>n>>m;
	init();
	for(int i=0;i<m;i++)
		cin>>a[i].u>>a[i].v>>a[i].w;
	kruskal();
	if(cnt==n-1) cout<<ans;
	else cout<<"orz";
	return 0;
}

prim ( 采用堆优化 和 链式前向星 ) 模板:

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;

struct node{
	int pos,dis;
	bool operator <(const node& x)const{
		return x.dis<dis;
	}
};
const int maxn=5001,maxm=200005<<1;
int n,m,cnt,k,ans;
int vv[maxm],ww[maxm],he[maxn],to[maxm];
void add_edge(int u,int v,int w){
	vv[++cnt]=v;
	ww[cnt]=w;
	to[cnt]=he[u];
	he[u]=cnt;
}
priority_queue<node> q;
int vis[maxn];
int dis[maxn];
void prim(){
	memset(dis,127,sizeof(dis));
	dis[1]=0;
	q.push((node){1,0});
	while(!q.empty()&&k<n){
		int x=q.top().pos;
		int w=q.top().dis;
		q.pop();
		if(vis[x]) continue;
		k++;
		vis[x]=1;
		ans+=w;
		for(int i=he[x];i;i=to[i]){
			if(ww[i]<dis[vv[i]]){
				dis[vv[i]]=ww[i];
				q.push((node){vv[i],ww[i]});
			}
		}
	}
}
int main()
{	
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		add_edge(u,v,w);
		add_edge(v,u,w);
	}
	prim();
	if(k==n) cout<<ans;
	else cout<<"orz";
}