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

Oscar_yuan

于 2024-07-15 21:52:42 发布

阅读量389

点赞数 12

文章标签： c++ 数据结构图论算法

本文链接：https://blog.csdn.net/2401_86305297/article/details/140449863

版权

【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 $N, M$ ，表示该图共有 $N$ 个结点和 $M$ 条无向边。

接下来 $M$ 行每行包含三个整数 $X_i,Y_i,Z_i$ ，表示有一条长度为 $Z_i$ 的无向边连接结点 $X_i,Y_i$ 。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模：

对于 $20\%$ 的数据， $N\le 5$ ， $M\le 20$ 。

对于 $40\%$ 的数据， $N\le 50$ ， $M\le 2500$ 。

对于 $70\%$ 的数据， $N\le 500$ ， $M\le 10^4$ 。

对于 $100\%$ 的数据： $1\le N\le 5000$ ， $1\le M\le 2\times 10^5$ ， $1\le Z_i \le 10^4$ 。

思路

本题需要并查集: 题解点这里

代码

#include<bits/stdc++.h>
using namespace std;
struct node{
    int from,to,c;
    node(int a,int b,int d){
        from=a,to=b,c=d;
    }
};
int fa[100010];
vector<node>v;
void init(int n){
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
}
bool cmp(node a,node b){
    return a.c<b.c;
}
int find(int x){
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
void Union(int x,int y){
    int xx=find(x);
    int yy=find(y);
    if(xx!=yy){
        fa[xx]=yy;
    }
}
int main(){
    int n,m;
    cin>>n>>m;
    init(n);
    for(int i=0;i<m;i++){
        int x,y,z;
        cin>>x>>y>>z;
        v.push_back(node(x,y,z));
    }
    long long sum=0;
    sort(v.begin(),v.end(),cmp);//对边权进行排序
    for(int i=0;i<m;i++){
        if(find(v[i].from)==find(v[i].to)) continue;//已有边
        sum+=v[i].c;//添加边
        Union(v[i].from,v[i].to);
    }
    for(int i=1;i<n;i++){
        if(find(fa[i])!=find(fa[i+1])){//判断是否连通
            cout<<"orz";
            return 0;
        }
    }
    cout<<sum;
}