Prime算法:
Prime算法的原理是贪心,通过对图上的点进行操作来产生最小生成树。
这个过程可以用两个集合的变化来表现出来,假设有一个集合U用来存放已加入最小生成树的点,另一个集合V存放未加入最小生成树的点。
因为要构造一棵最小生成树,只要两点之间可以相互到达就好,那么每次我们从V中取出来的点只要满足到U中某个点的距离比V中其他点到U中所有点的距离都要小,那么这个点就应该放进U中。为什么呢?因为每个点都要放进去才能构造出最小生成树,那么我们每次放进去的点与U中可以与它相连的点应该是最小的。
P3366 【模板】最小生成树(洛谷)
具体代码实现如下:
#include <iostream>
#include <algorithm>
#include <math.h>
#include <cstdio>
#include <cmath>
#include <stdlib.h>
#include <string.h>
using namespace std;
int n,m,from,to,w;
int G[5005][5005],dis[5005]; //G表示用邻接矩阵来存图,dis[i]表示集合中与i直接相连的点到i
bool vis[5005]; //的最小距离,vis[i]表示该点是否已在最小生成树集合中;
int Max=0x3f3f3f3f;
int Prime()
{
memset(dis,0x3f,sizeof(dis));//把当前不能到达的点初始化为无穷大;
dis[1]=0,vis[1]=1; //把1看成起点并收入最小生成树集合中(可随意找一个点作为起点);
int t=1,sum=0;
for(int i=1; i<=n-1; i++)
{
for(int j=1; j<=n; j++)
dis[j]=min(dis[j],G[t][j]); //更新最短路径中的点可以到达的点的距离;
t=-1;
for(int j=1; j<=n; j++) //找未加入的点中到已加入点距离最小的;
{
if(vis[j]==0&&(t==-1||dis[t]>dis[j]))
t=j;
}
if(dis[t]==Max) //找不到就直接返回;
return Max;
vis[t]=1; //把找到的点加入最短路径中;
sum+=dis[t];
}
return sum;
}
int main()
{
cin>>n>>m;
memset(G,0x3f,sizeof(G));
for(int i=1; i<=m; i++)
{
cin>>from>>to>>w;
G[from][to]=G[to][from]=min(G[from][to],w); //存图;
}
int ans=Prime();
if(ans==Max)
cout<<"orz"<<endl;
else
cout<<ans<<endl;
return 0;
}
Kruskal算法:
Kruskal算法通过处理边来得到最小生成树,其算法思想是贪心。
记录下图中的每一条边,然后每次取权值最小的边,同时把这条边的两个端点加入到最小生成树,使它们有同一个根节点,每次加入一条边之前要检查两个端点是否有相同的根节点,如果有相同的根节点是不能把这条边加入的,因为这回导致回路的产生,从而无法构成最小生成树。
P3366 【模板】最小生成树(洛谷)
具体代码实现如下:
#include <iostream>
#include <algorithm>
#include <math.h>
#include <cstdio>
#include <cmath>
#include <stdlib.h>
#include <string.h>
using namespace std;
int n,m,id,pre[5005]; //pre[i]表示i点的父节点;
struct node
{
int from,to,w;
} e[200005]; //用来保存每条边的信息;
bool cmp(node a,node b) //按权值给边排序;
{
return a.w<b.w;
}
int Find(int x) //找x的根节点(并查集);
{
if(x==pre[x])
return x;
else
return pre[x]=Find(pre[x]);
}
void initial() //pre初始化;
{
for(int i=1; i<=n; i++)
pre[i]=i;
}
void Kruskal()
{
initial(); //初始化父节点集合;
sort(e,e+id,cmp); //按权值由小到大排序;
int cnt=0,sum=0; //cnt记录最小生成树的边数,sum记录总权值;
for(int i=0; i<id; i++)
{
int u=e[i].from,v=e[i].to,w=e[i].w; //取边并查找端点的根节点;
int fu=Find(u),fv=Find(v);
if(fu==fv)
continue;
else
{
cnt++;
sum+=w;
pre[fu]=fv; //加入并查集;
}
}
if(cnt==n-1)
printf("%d\n",sum);
else
printf("orz\n");
}
int main()
{
cin>>n>>m;
for(int i=1; i<=m; i++)
{
int a,b,c;
cin>>a>>b>>c;
e[id++]= {a,b,c}; //存边;
}
Kruskal();
return 0;
}