P3366 【模板】最小生成树

最新推荐文章于 2025-12-09 11:28:31 发布

原创最新推荐文章于 2025-12-09 11:28:31 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法

最小生成树专栏收录该内容

7 篇文章

订阅专栏

本文主要探讨了如何利用C++编程解决洛谷P3366题。代码中运用了并查集数据结构，通过排序和合并操作，计算连通组件的最小代价。核心在于快速找到集合的根节点并进行合并。

在这里插入图片描述
https://www.luogu.com.cn/problem/P3366

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int p[N],n,m;
struct node{int a,b,c;};
bool cmp(node a,node b){return a.c<b.c;}
vector<node>ve; 
int find(int x)
{
	if(x!=p[x]) p[x]=find(p[x]);
	return p[x];
}
int f()
{
	int res=0,cnt=0;
	for(int i=1;i<=n;i++) p[i]=i;
	sort(ve.begin(),ve.end(),cmp);
	for(int i=0;i<ve.size();i++)
	{
		int a=ve[i].a,b=ve[i].b,c=ve[i].c;
		if(find(a)!=find(b))
		{
			p[find(a)]=find(b);
			res+=c;
			cnt++;
		}
	}
	if(cnt==n-1) return res;
	else return -1;
}
int main(void)
{
	cin>>n>>m;
	while(m--)
	{
		int a,b,c; cin>>a>>b>>c;
		ve.push_back({a,b,c});
	}
	int ans=f();
	if(ans==-1) puts("orz");
	else cout<<ans;
	return 0;
}