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

爱与愁大神

于 2024-08-03 15:54:31 发布

阅读量385

点赞数 8

文章标签： c++

本文链接：https://blog.csdn.net/2401_86025291/article/details/140891816

版权

题目描述

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

输入格式

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

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

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >;
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int N = 1e6+5;
int acc[N];
pair<int,pair<int,int>>va[N];//va[i]zhi di i tiao bian
int n,m;
int find(int x)
{
	if(acc[x]==x) return x;
	else return acc[x]=find(acc[x]);
}
void krukal()
{
	int cnt=0,sum=0;//sum   bian quan zong he,cnt bian de ge shu;
	for(int i=1;i<=n;i++) acc[i]=i;
	for(int i=1;i<=m;i++)
	{
		int a=va[i].se.fi,b=va[i].se.se,c=va[i].fi;
		int t1=find(a),t2=find(b);
		if(t1!=t2)
		{
			acc[t1]=t2;
			sum+=c;
			cnt++;
		}
	}
	if(cnt==n-1) cout<<sum;
	else cout<<"orz";
}
signed main()
{
    IOS;
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
    	int a,b,c;
    	cin>>a>>b>>c;
    	va[i]={c,{a,b}};
	}
	sort(va+1,va+1+m);
	krukal();
	return 0;
}

Xi,Yi。

输出格式

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

输入输出样例

输入 #1复制

输出 #1复制

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >;
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int N = 1e6+5;
int acc[N];
pair<int,pair<int,int>>va[N];//va[i]zhi di i tiao bian
int n,m;
int find(int x)
{
	if(acc[x]==x) return x;
	else return acc[x]=find(acc[x]);
}
void krukal()
{
	int cnt=0,sum=0;//sum   bian quan zong he,cnt bian de ge shu;
	for(int i=1;i<=n;i++) acc[i]=i;
	for(int i=1;i<=m;i++)
	{
		int a=va[i].se.fi,b=va[i].se.se,c=va[i].fi;
		int t1=find(a),t2=find(b);
		if(t1!=t2)
		{
			acc[t1]=t2;
			sum+=c;
			cnt++;
		}
	}
	if(cnt==n-1) cout<<sum;
	else cout<<"orz";
}
signed main()
{
    IOS;
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
    	int a,b,c;
    	cin>>a>>b>>c;
    	va[i]={c,{a,b}};
	}
	sort(va+1,va+1+m);
	krukal();
	return 0;
}