备战2023蓝桥国赛-重新理解Prim和Kruskal算法

Prim算法

题目描述：


解析：
prim算法跟dijkstra的朴素解法有很多相像部分，他们都是枚举中继节点来更新下一个节点，都是选最小的dist值来更新，时间复杂度都是O(n✖n)。

不同之处在于:
dijkstra的dist[i]表示1号点到i号点的最小距离，而prim的dist[i]表示用来更新dist[i]的中继节点与i点的最小距离。

所以prim算法每次更新时都会加上dist[t]，然后通过dist[t]来更新与之相连的节点的dist值。

那么为什么这样做可以算出最小生成树呢？
因为每次求出最小的dist[t]时，就代表着求出了用来更新dist[t]的中继节点与t点之间的最小值，又由于中继节点用完了就不能用了，如此循环n次就能求出最小生成树了。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
int n,m,g[N][N],dist[N],st[N];
int prim()
{
	memset(dist,0x3f,sizeof(dist));
	int res=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!st[j]&&(t==-1||dist[t]>dist[j]))
			t=j;
		}
		if(i&&dist[t]==INF) return INF;
		if(i) res+=dist[t];
		for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
		st[t]=1;
	}
	return res;
}
int main()
{
	scanf("%d%d",&n,&m);
	memset(g,0x3f,sizeof(g));
	while(m--)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		g[u][v]=g[v][u]=min(g[u][v],w);
	}
	int t=prim();
	if(t==INF) printf("impossible");
	else printf("%d",t);
	return 0;
}

Kruskal算法

题目描述：


解析：
Kruskal算法跟dijkstra的堆优化版有点类似，都是枚举边，但写法上有很大不同，dijkstra是用单链表来存边的，Kruskal则是用结构体来存，而且枚举边的方法也有很大的不同，Kruskal单纯地从小到大枚举边，而堆优化版的dijkstra则是用到了优先队列来存边和点，然后从中选最小的dist值，情况更复杂；

当然，由于题目性质不同，这两个解法肯定不同，接下来就说Kruskal的具体解法：
先是用结构体存边，再对边进行从小到大的排序，然后枚举边，判断边上的两个节点是否在同一个连通块里，不在就把他们连起来，res加上边的值，cnt++(res用来统计权值之和，cnt用来统计连通块中边的数目)，如此循环m次，判断cnt是否等于n-1，是则能求出最小生成树，否则就不存在。

那么为什么这种解法可以求出最小生成树呢？
因为每次循环，如果两个点不在同一个连通块里，就能确定两个点之间的最小值，如此循环m次，如果存在最小生成树的话，就一定能求出能将所有点相连的具有n个点，n-1条边的生成树，而且是最小的。

代码：

#include<bits/stdc++.h>
using namespace std;
const int M=2e5+10,N=1e5+10,INF=0x3f3f3f3f;
int n,m,p[N];
struct node{
	int a,b,w;
}edges[M];
int find(int x){
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
bool cmp(node x,node y)
{
	return x.w<y.w;
}
int kruskal()
{
	sort(edges,edges+m,cmp);
	
	for(int i=1;i<=n;i++) p[i]=i;
	
	int res=0,cnt=0;
	for(int i=0;i<m;i++)
	{
		int a=edges[i].a,b=edges[i].b,w=edges[i].w;
		a=find(a),b=find(b);
		if(a!=b)
		{
			p[a]=b;
			res+=w;
			cnt++; 
		}
	}
	if(cnt==n-1) return res;
	else return INF;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++)
	{
		int a,b,w;
		scanf("%d%d%d",&a,&b,&w);
		edges[i]={a,b,w};
	}
	int t=kruskal();
	if(t==INF) printf("impossible");
	else printf("%d",t);
}