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

最小生成树

题目描述

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

输入格式

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

接下来 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

说明/提示

数据规模：

对于 20% 的数据，N≤5，M≤20。

对于 40% 的数据，N≤50，M≤2500。

对于 70% 的数据，N≤500，M≤10^4。

对于 100% 的数据：1≤N≤5000，1≤M≤2×10^5，1≤Zi​≤10^4。

样例解释：

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

代码： 

#include <bits/stdc++.h>
using namespace std;
int n,m,sum,cnt;
pair<int,pair<int,int>> sn[1000010];//开到2e5就够 
int anc[100010];//开到5000就够 
int find(int x){//并查集模板 
	if(x!=anc[x]) anc[x]=find(anc[x]);
	return anc[x];
}
void kruskal(){
	for(int i = 1;i<=n;i++) anc[i]=i;
	for(int i = 1;i<=m;i++){
		int a=sn[i].second.first,b=sn[i].second.second,c=sn[i].first;
		int t1=find(a),t2=find(b);
		if(t1!=t2){
			anc[t1]=t2;
			sum+=c;//边权和 
			cnt++;//边的条数 
		}
	}
	if(cnt==n-1) cout << sum;//n个点连成一棵树需要n-1条边 
	else cout << "orz";
} 
int main(){
	cin >> n >> m;
	for(int i = 1;i<=m;i++){
		int x,y,z;
		cin >> x >> y >> z;
		sn[i]={z,{x,y}};//z是边权,x,y是边的两个点 
	}
	sort(sn+1,sn+1+m);//sort先按第一个排
	kruskal(); 
	return 0;
}