图论——最小生成树（prim算法和Kruskal算法）模板

最新推荐文章于 2022-11-28 20:41:24 发布

Asuka_46_

最新推荐文章于 2022-11-28 20:41:24 发布

阅读量232

点赞数 3

分类专栏：算法文章标签： prim kruskal 图论 acm竞赛 c++

本文链接：https://blog.csdn.net/qq_33969563/article/details/114593653

版权

算法专栏收录该内容

25 篇文章 0 订阅

订阅专栏

洛谷 P3366 【模板】最小生成树

在这里插入图片描述

Kruskal算法（贪心思想）

使用Kruskal算法求最小生成树时，建图用的edge数组中存三个量，x,y,len,代表x和y之间有一条长度为len的边
算法过程：
1.把这m条边以边长从小到大排序
2.每次取最小的那条边，判断这条边相连的两个点是否已经在树上了（利用并查集），如果没在加上这条边，把这两个点加到树上去（合并两个点）。
AC代码：

#include <bits/stdc++.h>

using namespace std;

struct node
{
    int x, y, len;
}edge[200050];
int n, m;
int ans, cnt;
int fx, fy;
int fa[5050];

int findfa(int x)
{
    return fa[x] == x ? x : (fa[x] = findfa(fa[x]));
}

bool cmp(struct node A,struct node B)
{
    return A.len < B.len;
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) fa[i] = i;
    for(int i = 1; i <= m; ++i) scanf("%d%d%d", &edge[i].x, &edge[i].y, &edge[i].len);
    sort(edge + 1, edge + m + 1, cmp);
    for(int i = 1; i <= m; ++i)
    {
        fx = findfa(edge[i].x);
        fy = findfa(edge[i].y);
        if(fx != fy)
        {
            fa[fy] = fx;
            ++cnt;
            ans += edge[i].len;
            if(cnt >= n - 1)    break;
            //cout << ans << endl;
        }
    }
    if(cnt < n - 1) cout << "orz\n";
    else    printf("%d\n", ans);

    return 0;
}

prim算法

维护一个数组，dist[i]代表点i到树上点的最小距离

先把dist数组赋为无穷大，然后任意取一个点，假设我们都取第一个点，然后将dist[1]置为0，然后把所有与第一个点相连的边加到优先队列里去，之后每次取队首，判断与这条边相连的那个点是否已经在树上了，没有则加上去，注意这里加到队列里的结构体只有两个元素有用，并且这两个元素的意义有变化，t代表这个点，l代表这个点到树的最小距离（即dist[i]）,再将所有于这个点相连的边加到优先队列里去，循环操作。(一直再说把边加到了队列里，其实加的不是边，结构体两个元素的意义点和点到树的距离)

AC代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

int n, m, x, y, z, tot;
ll ans;
int head[5050];
struct node
{
    int to, l, next;
    bool operator < (const node &a) const
    {
        return l > a.l;
    }
}edge[400050];
int dist[5050];
priority_queue<node> q;

void addedge(int u, int v, int z)
{
    edge[++tot].to = v;
    edge[tot].l = z;
    edge[tot].next = head[u];
    head[u] = tot;
}

void prim()
{
    ans = 0;
    node t1, t2;
    //先把dist数组置为无穷大
    memset(dist, 127, sizeof(dist));
    dist[1] = 0;//取第一个点，把所有与第一个点相连的边入队
    for(int i = head[1]; i != 0; i = edge[i].next)
    {
        if(edge[i].l < dist[edge[i].to])
        {
            dist[edge[i].to] = edge[i].l;
            q.push(edge[i]);
        }
    }
    while(!q.empty())
    {
        //取队首，弹出，如果这条边连的那个点已经在树上了就continue
        t1 = q.top();
        q.pop();
        if(dist[t1.to] == 0) continue;
        //否则加到树上去，标记这个点
        ans += t1.l;
        dist[t1.to] = 0;
        //将与这个点相连的所有边加到队列里
        for(int i = head[t1.to]; i != 0; i = edge[i].next)
        {
            //找到这条边所连的那个点，判断这个点是否已经在树上了
            int u = edge[i].to;
            if((dist[u] != 0) && (dist[u] > edge[i].l))
            {
                dist[u] = edge[i].l;
                t2.to = u;
                t2.l = dist[u];
                q.push(t2);
            }
        }
    }
    printf("%lld\n", ans);
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= m; ++i)
    {
        scanf("%d%d%d", &x, &y, &z);
        addedge(x, y, z);
        addedge(y, x, z);
    }
    prim();
    return 0;
}