挖沟

挖沟

https://ac.nowcoder.com/acm/problem/17509

最小生成树模板题

直接上代码
大概意思解释一下,
1.首先最小生成树的克鲁斯卡尔算法相当于并查集的升级版,与并查集不同的是,他要对边的权值进行排序,并且最后生成树的边是m-1条
2.我们用一个结构体内嵌比较函数去记录和排序边,外加一个路径压缩和一个判断联通的函数
3.然后我们对已经排好队的边进行加边操作,如果两个边没有被联通,那我们就加边,不断地记录这个操作,最后得出来的就是最小生成树的最小权值之和

方法一

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
int pre[maxn];
struct node
{
    int u, v, e;
    bool operator<(const node x)
    {
        return e < x.e;
    }
} w[maxn];
int find(int x)
{
    return pre[x] == x ? x : pre[x] = find(pre[x]);
}
int join(int x, int y)
{
    int a = find(x);
    int b = find(y);
    if (a != b)
    {
        pre[a] = b;
        return 1;
    }
    return 0;
}
int main()
{
    int n, m, ans = 0;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
        pre[i] = i;
    for (int i = 1; i <= m; ++i)
        scanf("%d%d%d", &w[i].u, &w[i].v, &w[i].e);
    sort(w + 1, w + m + 1);
    for (int i = 1; i <= m; ++i)
    {
        if (join(w[i].u, w[i].v))
            ans += w[i].e;
    }
    printf("%d\n", ans);
}

方法二

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
int pre[maxn];
struct node
{
    int u, v, e;
    bool operator<(const node x)
    {
        return e < x.e;
    }
} w[maxn];
int find(int x)
{
    return pre[x] == x ? x : pre[x] = find(pre[x]);
}
void join(int x, int y)
{
    pre[find(x)] = find(y);
}
int main()
{
    int n, m, ans = 0;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
        pre[i] = i;
    for (int i = 1; i <= m; ++i)
        scanf("%d%d%d", &w[i].u, &w[i].v, &w[i].e);
    sort(w + 1, w + m + 1);
    for (int i = 1; i <= m; ++i)
    {
        if (find(w[i].u) != find(w[i].v))
        {
            join(w[i].u, w[i].v);
            ans += w[i].e;
        }
    }
    printf("%d\n", ans);
}
牛客课后习题题解 文章被收录于专栏

等待蜕变

全部评论

相关推荐

黑皮白袜臭脚体育生:可以看看我的开源仿b站前后端分离微服务项目,技术栈相当先进,符合企业校招需求,具体为springboot security, nacos,openfeign,gateway,redis,elasticsearch,rocketmq,minio,mybatis-plus,mybatis-plus-join,druid,jwt,swagger,gson,hutool,websocket,讯飞星火api,jave,xxl-job,zipkin,slueth,可快速下载所有用到的中间件和远程连接中间件软件而不用麻烦的去官网找包以及只需小改存放路径就可缓存前端静态资源的nginx和前端dist包,无需会任何前端极速实现本机运行前端,所有文档教程只在牛客,有各中间件启动教程,有配套简历写法速成简历,github已经330star
点赞 评论 收藏
分享
2024-11-18 09:44
Java
小白也想要offer:简历别放洋屁,搞不还还放错了,当然你投外企除外,以上纯属个人观点
点赞 评论 收藏
分享
评论
2
收藏
分享
牛客网
牛客企业服务