题目描述
如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz
输入输出格式
输入格式：
第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）
接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi
输出格式：
输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
struct edge{
ll u,v,w;
}e[200000+10];
bool cmp(const edge &a,const edge &b)
{
return a.w<b.w;
}
ll f[200005],n,m,ans,cnt;
ll find(ll x)
{
if(f[x]!=x)
f[x]=find(f[x]);
return f[x];
}
void kruskal()
{
sort(e,e+m,cmp);
for(int i=0;i<m;i++)
{
ll u=e[i].u, v=e[i].v, w=e[i].w;
ll fu=find(u), fv=find(v);
if(fu==fv) continue;
ans+=w;
f[fv]=fu; cnt++;
if(cnt==n-1) break;
}
}
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
{
f[i]=i;
}
for(int i=0;i<m;i++)
{
scanf("%lld %lld %lld",&e[i].u,&e[i].v,&e[i].w);
}
kruskal();
if(cnt!=n-1)
{
printf("orz");
}
printf("%lld",ans);
return 0;
}