洛谷-3366 【模板】最小生成树

题目描述
如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz
输入输出格式
输入格式：
第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）
接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi
输出格式：
输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例
输入样例#1：
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出样例#1：
7

解释：都是经典题目

#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 200004
#define V 5003
using namespace std;
struct edge{
    int a,b,c;
};
edge E[N];
int father[V]={0};
void init(){
    for(int i=0;i<V;i++) father[i]=i;
}
int find(int x){
    if(father[x]==x) return x;
    return father[x]=find(father[x]);
}
void merge(int x,int y){
    x=find(x);y=find(y);
    father[x]=y;
}
bool same(int x,int y){
    return find(x)==find(y);
}
bool cmp(edge &x,edge &y){
    return x.c<y.c;
}
int n=0,m=0;
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m;
    init();
    int ret=0;
    int num=0;
    for(int i=1;i<=m;i++) cin>>E[i].a>>E[i].b>>E[i].c;
    sort(E+1,E+1+m,cmp);
    for(int i=1;i<=m;i++){
        if(!same(E[i].a,E[i].b)){
            merge(E[i].a,E[i].b);
            ret+=E[i].c;
            num++;
        }
    }
    if(num<n-1) cout<<"orz"<<endl;
    else cout<<ret<<endl;
    return 0;
}