最小生成树模板（Kruskal算法和prim算法）

Kruskal算法模板

洛谷模板题目链接：https://www.luogu.com.cn/problem/P3366

Kruskal算法将一个连通块当做一个集合，先对所有边进行一次排序，然后按顺序枚举每一条边，当此条边的两个点满足不属于同一个集合的时候，如果这条边连接的两个点属于同一集合，就跳过。直到选取了n-1条边为止。

这里实用并查集对集合进行维护。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<map>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#define ll long long
#define inf 0x3f3f3f3f
#define llinf 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 200005;
int n,m;
int f[maxn];
//vector<int> g[maxn];
struct edge{
	int u,v,w;
}a[maxn];

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

bool cmp(edge d,edge e){
	return d.w<e.w;
} 

ll ans=0;

int main()//Kruskal算法 
{
	cin>>n>>m;
	int u,v,z;
	for(int i=1;i<=n;i++)f[i]=i;
	for(int i=1;i<=m;i++){
		cin>>a[i].u>>a[i].v>>a[i].w;
	}
	int cnt=0;
	sort(a+1,a+1+m,cmp);
	int fx,fy;
	for(int i=1;i<=m;i++){
		fx=find(a[i].u);
		fy=find(a[i].v);
		if(fx!=fy)
		{
			f[fx]=fy;ans+=a[i].w;cnt++;
		}
		if(cnt==n-1)break;
	}
	cout<<ans;
	return 0;
}

Prim算法模板

和kruskal算法相似的贪心思想，不同的是，prim算法适合于稠密图

不同于Kruskal算法的每次向集合中加入边，prim算法每次向集合中加入边，所以引用vis判断是否已进入集合，dis[u]表示不在集合的点u到集合中的最短长度

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<map>
#include<algorithm>
#include<queue>
#include<set>
#define ll long long
#define inf 0x3f3f3f3f
#define llinf 0x3f3f3f3f3f3f3f3f
#define maxm 200005
#define maxn 5005
using namespace std;
struct edge{
	int v,w,next;
}e[maxm<<1];
int head[maxn],dis[maxn],cnt,n,m,tot,now=1,ans;

bool vis[maxn];

void adde(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}

inline void init(){
	cin>>n>>m;
	for(int i=1,u,v,w;i<=m;i++){
		cin>>u>>v>>w;
		adde(u,v,w);adde(v,u,w);
	}
} 


typedef pair <int,int> pii;
priority_queue <pii,vector<pii>,greater<pii> > q;
ll sum=0;
inline void prim(){
	
	dis[1]=0;
	q.push(make_pair(0,1));
	while(!q.empty()&&sum<n){
		int d = q.top().first,u=q.top().second;
		q.pop();
		if(vis[u])continue;
		sum++;
		ans+=d;5
		vis[u]=1;
		for(int i=head[u];i!=-1;i=e[i].next){
			if(e[i].w<dis[e[i].v]){
				dis[e[i].v]=e[i].w;
				q.push(make_pair(dis[e[i].v],e[i].v));
			}
		}
	}
	
}

int main()//prim算法 
{
	memset(dis,127,sizeof(dis));
	memset(head,-1,sizeof(head));
	init();
	prim();
	cout<<ans;
	return 0;
}