关于最小生成树的基础模版

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

// 定义边的结构体
struct Edge {
    int u, v, w; // 边的起点、终点和权重
};

// 比较函数,用于排序边
bool operator<(const Edge&amp; a, const Edge&amp; b) {
    return a.w < b.w;
}

// 并查集的父节点数组
int parent[100005];

// 查找根节点
int find(int x) {
    return parent[x] = find(parent[x]);
}

// 合并两个集合
void merge(int x, int y) {
    int px = find(x), py = find(y);
    if (px != py) parent[px] = py;
}

int main() {
    int n, m; // n为顶点数,m为边数
    cin >> n >> m;
    
    // 读取所有边
    vector<Edge> edges(m);
    for (int i = 0; i < m; i++) {
        cin >> edges[i].u >> edges[i].v >> edges[i].w;
    }
    
    // 按照边的权重排序
    sort(edges.begin(), edges.end());
    
    // 初始化并查集
    for (int i = 0; i <= n; i++) parent[i] = i;
    
    long long ans = 0; // 最小生成树的总权重
    int cnt = 0; // 已经加入最小生成树的边数
    
    // 遍历所有边
    for (auto&amp; e : edges) {
        int pu = find(e.u), pv = find(e.v);
        if (pu != pv) {
            merge(pu, pv);
            ans += e.w;
            cnt++;
        }
        if (cnt == n - 1) break; // 如果已经加入 n-1 条边,停止
    }
    
    // 输出结果
    if (cnt == n - 1) {
        cout << ans << endl;
    } else {
        cout << &quot;IMPOSSIBLE&quot; << endl; // 如果无法形成最小生成树
    }

    return 0;
}
全部评论

相关推荐

04-02 10:09
门头沟学院 Java
用微笑面对困难:这里面问题还是很多的,我也不清楚为啥大家会感觉没啥问题。首先就是全栈开发实习9个月的内容都没有java实习生的内容多,1整个技术栈没看出太核心和难点的内容,感觉好像被拉过去打杂了,而且全栈基本上很容易被毙。里面能问的bug是在太多了比如L:继承 BaseMapper 可直接使用内置方法’。请问你的 BaseMapper 是如何扫描实体类注解如果瞬时产生 100 个上传任务,MySQL 的索引设计是否会有瓶颈?你做过分库分表或者索引优化吗?全栈的内容可以针对动态难点去搞,技能特长写在下面吧,你写了这么多技能,项目和实习体现了多少?你可以在项目里多做文章然后把这个放下去,从大致来看实习不算太水,有含金量你也要写上内容针对哨兵里面的节点变化能问出一万个问题,这个很容易就爆了。
提前批简历挂麻了怎么办
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务