最小生成树（模板）

最新推荐文章于 2024-03-24 14:27:30 发布

算法编程张老师

最新推荐文章于 2024-03-24 14:27:30 发布

阅读量221

点赞数 6

文章标签：算法数据结构

本文链接：https://blog.csdn.net/m0_57071296/article/details/119792923

版权

该博客讨论了如何解决无向图的最小生成树问题，并在图不连通时输出特殊标识。文章介绍了一种算法，通过Kruskal或Prim算法找到最小生成树，并检查图是否连通。当图连通时，算法会输出最小生成树的总权重；反之，输出'orz'。题目给出了具体的输入输出样例和数据规模限制，包括不同测试用例的节点和边数量，以及边的权重范围。

摘要由CSDN通过智能技术生成

题目描述

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

输入格式

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

接下来 M 行每行包含三个整数 Xi,Yi,Zi，表示有一条长度为 Zi 的无向边连接结点 Xi,Yi。

输出格式

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

输入输出样例

输入 #1复制

输出 #1复制

说明/提示

数据规模：

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

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

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

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

样例解释：

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

#include<bits/stdc++.h>
using namespace std;
int p[310];
struct edge{
	int a,b,w;
	bool operator<(const edge&a)const{
	return w<a.w;//小到大 
	}
}s[10010];
int find(int x){//根节点 
	
	if (x!=p[x]) return find(p[x]);
	return p[x]; 
}
int main(){
	int n,m,k;
	cin>>m>>n;
		if(n==0)
            return 0;
		for(int i=0;i<n;i++){//几条路 
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);//c代表权值 
			s[i]={a,b,c};//结构体写法 
	      }
	sort(s,s+n,cmp);//排列方式按照结构体里面的排列来 就是小到大 
	for(int i=1;i<310;i++){
		p[i]=i;//自己是自己的根节点 
	}
	int num=0,maxn=0;
	for(int i=0;i<n;i++){
		int a1=find(s[i].a);
		int b1=find(s[i].b);
		if(a1!=b1){
			p[a1]=b1;
			maxn=maxn+s[i].w;
		}
	}
	cout<<maxn<<endl;//几条路和最大值 
	return 0;//听懂掌声
}