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

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

洛谷传送门

题目描述

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

输入格式

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

接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iX**i,Y**i,Z**i，表示有一条长度为 Z_iZ**i 的无向边连接结点 X_i,Y_iX**i,Y**i。

输出格式

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

---

题解：

代码：

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=5010;
const int maxm=2e5+5;
int n,m,ans,cnt;
int fa[maxn];
struct edge
{
	int x,y,z;
}e[maxm];
bool cmp(edge a,edge b)
{
	return a.z<b.z;
}
int find(int x)
{
	return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
		scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].z);
	sort(e+1,e+m+1,cmp);
	for(int i=1;i<=n;i++)
		fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int x=e[i].x,y=e[i].y,z=e[i].z;
		int fx=find(x);
		int fy=find(y);
		if(fx!=fy)
		{
			fa[fx]=fy;
			ans+=z;
			cnt++;
		}
		if(cnt==n-1)
		{
			printf("%d\n",ans);
			return 0;
		}
	}
	printf("orz\n");
	return 0;
}