最小生成树
一、Prim求最小生成树
Prim最小生成树的原理是将一 点与集合中距离最近的点连接,在将其加入集合中,循环此操作。
代码与dijkstrs算法类似,如下:
const int INF = 0x3f3f3f3f;
int dist[N],g[N][N];
bool vis[N];
int prim()
{
memset(dist,0x3f,sizeof dist);//dist储存的是每个点到集合的距离,先将其初始化为无限大
int ans = 0;//存储答案
for(int i = 0;i < n;i++)
{
int t = -1;
for(int j = 1;j <= n;j++)//找出距离集合最小的点
if(!vis[j] && (t == -1 || dist[t] > dist[j]))
t = j;
if(i && dist[t] == INF)return INF;//如果不是第一个点,且距离集合的距离为无限大,说明不存在生成树
if(i)ans += dist[t];
for(int j = 1;j <= n;j++)dist[j] = min(dist[j],g[t][j]);//更新每个点到集合的距离
}
}
二、Kruskal求最小生成树
Kruskal算法结合了并查集的原理,按边权从小到大排列,然后进行连接并计算边权和,以求最小生成树。
首选是利用类或者结构体储存参数,并重构<运算符以使用sort函数:
class Edge
{
publice:
int a,b,c;
bool operator<(const Edge &op)const
{
return c < op.c;
}
}edge[N];
在将数据读入之后,利用sort函数将边权从小到大进行排序,最后就是调用Kruckal函数了:
int kruskal()
{
for(int i = 1;i <= n;i++) f[i] = i;//并查集
sort(edge,edge + m);
itn ans = 0,cnt = 0;
for(int i = 0;i < m;i++)
{
int a = edge[i].a,b = edge[i].b,c = edge[i].c;
a = find(a),b = find(b);
if(a != b)
{
f[a] = b;
ans += c;
cnt++;
}
}
if(cnt < n - 1)return INF;//定义常量INF = 0x3f3f3f3f
return ans;
}
最后是find函数,具体的可参考这篇文章:并查集的原理与应用_Kicamon的博客-CSDN博客
int find(int a)
{
if(f[a] != a) f[a] = find(f[a]);
return f[a];
}