【luogu3366】 最小生成树模板

(http://www.elijahqi.win/2017/07/11/%E3%80%90luogu3366%E3%80%91-%E6%9C%80%E5%B0%8F%E7%94%9F%E6%88%90%E6%A0%91%E6%A8%A1%E6%9D%BF/)
题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出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
说明

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=20

对于40%的数据：N<=50，M<=2500

对于70%的数据：N<=500，M<=10000

对于100%的数据：N<=5000，M<=200000

样例解释：

所以最小生成树的总边权为2+2+3=7

#include<cstdio>
#include<algorithm>
#define N 5500
#define M 220000
using namespace std;
int num,n,fa[N];
inline int read(){
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') {
        x=x*10+ch-'0';ch=getchar();
    }
    return x;
}
struct node{
    int x,y,w;
}data[M];
inline void insert1(int x,int y,int w){
    data[++num].x=x;data[num].y=y;data[num].w=w;
}
inline int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
inline bool cmp(node a,node b){
    return a.w<b.w;
}
int main(){
    freopen("3366.in","r",stdin);
    freopen("3366.out","w",stdout);
    n=read();num=0;int m=read();
    int x,y,z;
    for (int i=1;i<=m;++i) x=read(),y=read(),z=read(),insert1(x,y,z);
    for (int i=1;i<=n;++i) fa[i]=i;
    int tot=0;
    sort(data+1,data+1+num,cmp);
    //for (int i=1;i<=num;++i )printf("%d %d %d\n",data[i].x,data[i].y,data[i].w);
    int i;
    long long ans=0;
    for (i=1;i<=num;++i){
        int p=find(data[i].x),q=find(data[i].y);
        if (p!=q) fa[p]=q,++tot,ans+=data[i].w;
    //    printf("%d %d %d\n",data[i].x,data[i].y,data[i].w);
        if (tot==n-1) break;
    }
    if (tot<n-1) printf("orz");else
    printf("%lld",ans);
    return 0;
}