最小生成树

题目描述

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

 

#include<bits/stdc++.h>
using namespace std;
int f[5005],n,m,ans,ex,ey,cnt;
struct node
{
	int x,y,z;
}a[200005];
bool cmp(node a,node b)
{
	return a.z<b.z;
}
int find(int x)
{
	while(x!=f[x])
	{
		x=f[x]=f[f[x]];
	}
	return x;
}//找到他最终的爸爸 
void kruskal()
{
	sort(a+1,a+m+1,cmp);//按权值的大小排序 
	for(int i=1;i<=m;i++)
	{
		ex=find(a[i].x);
		ey=find(a[i].y);//找到他们的爸爸 
		if(ex==ey) continue;//如果他们已经连接在一起了 有个一个共同的爸爸 就不用这条边了 
		ans+=a[i].z;//用这条边 加上权值 
		f[ey]=ex;//把这俩个个体连接一起来 
		if(++cnt==n-1) break; //如果找到了n-1条边就代表所有的节点都连接在了一起 
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++) cin>>a[i].x>>a[i].y>>a[i].z;
	for(int i=1;i<=n;i++) f[i]=i;//令每个人的爸爸都是自己 都是独立的个体 
	kruskal();
	cout<<ans<<endl;
	return 0;
}