生成树prim与kruskal算法代码详解

Prim算法代码

prim 总的代码思路类似于 迪杰斯特拉

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 110, INF = 0x3f3f3f3f;

int n,m;
int g[N][N];

void prim()
{
    int dis[n+1], vis[n+1];//dis[i]是指点 i 到最小生成树的距离， vis[i]标记i点是否在最小生成树里
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i<=n; i++)//假设当前最小生成树中只有 点1(起点)
        dis[i] = g[1][i];
    vis[1] = 1;
    int ans = 0;
    for(int k = 1; k<n; k++)//只要进行 n - 1 轮
    {
        int u, min_dis = INF;
        for(int i = 1; i<=n; i++)//找到离最小生成树最近的点 u
            if(min_dis>dis[i] && !vis[i])//也可以不用!vis[i]，直接用dis[i]更简洁
                min_dis = dis[u=i];
        vis[u] = 1;
        dis[u] = 0;//把u点加入到最小生成树，那么u点到最小生成树的距离就是0
        ans += min_dis;
        for(int i = 1; i<=n; i++)
            if(dis[i]>g[i][u])//i点到最小生成树的距离是否比i点到u点的距离小，如果是，就可以更新i点到最小生成树的距离
                dis[i] = g[i][u];
    }
    printf("%d\n",ans);
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        int u,v,w;
        for(int i = 0; i<=n; i++)
            for(int j = 0; j<=n; j++)
                g[i][j] = INF;
        for(int i = 0; i<m; i++)
        {
            scanf("%d%d%d",&u, &v, &w);
            g[u][v] = g[v][u] = w;
        }
        prim();
    }
    return 0;
}

Kruskal算法代码

运用了并查集思想

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

const int N = 110;

int n,m;
int root[N];

struct node
{
    int u, v, w;
}g[N*N];

void init()
{
    for(int i = 0; i<=100; i++)
        root[i] = i;
}

int find_(int x)
{
    return x == root[x] ? x : root[x] = find_(root[x]);
}

bool cmp(node a, node b)
{
    return a.w<b.w;
}

void kruskal()
{
    int ans = 0;
    init();
    sort(g, g+m, cmp);//把所有边按权从大到小排序
    for(int i = 0; i<m; i++)
    {
        //每遍历一条边，就判断两点是否在同一个集合中；
        //如果不是，把两个集合连起来，并且答案加上边权
        //由于是按边权从小到大遍历的，所以最终得到的集合一定是最小生成树
        int ra = find_(g[i].u);
        int rb = find_(g[i].v);
        if(ra!=rb)
        {
            root[ra] = rb;
            ans += g[i].w;
        }
    }
    printf("%d\n",ans);
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i = 0; i<m; i++)
        {
            scanf("%d%d%d",&g[i].u, &g[i].v, &g[i].w);
        }
        kruskal();
    }
    return 0;
}