【模板】最小生成树-kruskal

int father[5010], n, m;
int find(int x)//找根函数,记得进行路径压缩
{
    if(father[x] == x) return x;
    else return father[x] = find(father[x]);
}
int same(int x, int y)//简化代码
{
    if(find(x) == find(y)) return 1;
    else return 0;
}
struct edge
{
    int u, v, cost;//边的两个端点,cost是边的长度
}e[200010];
int cmp(edge e1, edge e2)
{
    return e1.cost < e2.cost;
}
int kruskal()
{
    sort(e + 1, e + 1 + m, cmp);
    int res = 0;//边权总和
    int cnt = 0;//边的数量
    for(int i = 1; i <= m; i++)
    {
        if(same(e[i].u, e[i].v) == 0)
        {
            cnt++;
            res += e[i].cost;
            father[find(e[i].u)] = find(e[i].v);
        }
    }
    if(cnt == n - 1) return res;
    else return -1;
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        father[i] = i;
    }
    for(int i = 1; i <= m; i++)
    {
        cin >> e[i].u >> e[i].v >> e[i].cost;
    }
    int ans = kruskal();
    if(ans == -1) cout << "orz\n";
    else cout << ans << '\n';
    return 0;
}
posted @   joylz  阅读(6)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
点击右上角即可分享
微信分享提示