最小生成树

最小生成树

一、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];
}