最小生成树prim(学习笔记）

Kruskal就像是往图中填边。

而prim就像是这棵树在慢慢长大。

prim：将距离树最近的顶点加入的树中，然后更新其余顶点距离树的距离。重复操作，直到所有的顶点都在树中。

这与dijkstra：将距离源点最近的顶点的边变为确定值（不再改变的值），然后更新其余顶点到源点的距离。重复操作，直到所有的边变为确定值。

十分类似。

在dij中用dis[k]记录源点到各个顶点的距离,假设最近点为j，满足dis[k]>dis[j]+e[j][k]则更新dis[k]。

prim则是用dis[k]距离各个顶点到树的距离，满足dis[k]>e[j][k],则更新dis[k]。

---

P3366 【模板】最小生成树 https://www.luogu.com.cn/problem/P3366

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 N,M，表示该图共有 N 个结点和 M 条无向边。

接下来 M 行每行包含三个整数 Xi​,Yi​,Zi​，表示有一条长度为 Zi​ 的无向边连接结点 Xi​,Yi​。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 #1复制

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出 #1复制

7

说明/提示

数据规模：

对于 20% 的数据，N≤5，M≤20。

对于 40% 的数据，N≤50，M≤2500。

对于 70% 的数据，N≤500，M≤10^4。

对于 100% 的数据：1≤N≤5000，1≤M≤2×10^5，1≤Zi​≤10^4。

样例解释：

所以最小生成树的总边权为 2+2+3=7。

代码实现

邻接矩阵

#include<bits/stdc++.h>
using namespace std;
int inf=1000000000;
int e[10000+5][10000+5];
int dis[10000+5];
int book[10000+5];
int main()
{
	int n,m;
	long long ans=0;
	scanf("%d%d",&n,&m);
	int u,v,w;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			e[i][j]=inf;
		}
		e[i][i]=0;
	}	
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		if(w<e[u][v]) 
		{
			e[u][v]=w;//无向图
			e[v][u]=w;
		}
	}
	for(int i=2;i<=n;i++)
	{
		dis[i]=e[1][i];
	}
	dis[1]=0;
	book[1]=1;
	int near;
	for(int i=2;i<=n;i++)
	{
		int minn=inf;
		for(int j=1;j<=n;j++)
		{
			if(book[j]==0&&dis[j]<minn)
			{
				minn=dis[j];
				near=j;
			}
		}
		book[near]=1;
		for(int k=2;k<=n;k++)
		{
			if(book[k]==0&&dis[k]>e[near][k])
			{
				dis[k]=e[near][k];
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
		if(dis[i]==inf)
		{
			printf("orz");
			return 0;
		}
		else
		{
			ans+=dis[i];
		}
	}
	printf("%lld",ans);
	return 0;
}

---

还可以通过邻接表，优化内存。

代码实现

#include<bits/stdc++.h>
using namespace std;
int inf=1000000000;
int u[400000+5],v[400000+5],w[400000+5];
int dis[10000+5];
int book[10000+5];
int first[10000+5];
int nex[400000+5];
int main()
{
	int n,m;
	long long ans=0;
	scanf("%d%d",&n,&m);
	for(int i=0;i<=n;i++)
	{
		first[i]=-1;
		dis[i]=inf;
	} 
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u[i],&v[i],&w[i]);
		nex[i]=first[u[i]];
		first[u[i]]=i;
		u[i+m]=v[i];
		v[i+m]=u[i];
		w[i+m]=w[i];
		nex[i+m]=first[v[i]];
		first[v[i]]=i+m;
	}
	int k=first[1];
	while(k!=-1)
	{
		if(w[k]<dis[v[k]]) dis[v[k]]=w[k];
		k=nex[k];
	}
	dis[1]=0;
	book[1]=1;
	int near;
	for(int i=2;i<=n;i++)
	{
		int minn=inf;
		for(int j=1;j<=n;j++)
		{
			if(book[j]==0&&dis[j]<minn)
			{
				minn=dis[j];
				near=j;
			}
		}
		book[near]=1;
		k=first[near];
		while(k!=-1)
		{
			if(book[v[k]]==0&&dis[v[k]]>w[k])
			{
				dis[v[k]]=w[k];
			}
			k=nex[k];
		}
	}
	for(int i=1;i<=n;i++)
	{
		if(dis[i]==inf)
		{
			printf("orz");
			return 0;
		}
		else
		{
			ans+=dis[i];
		}
	}
	printf("%lld",ans);
	return 0;
}

---

还可以堆优化