蒟蒻刚学1小时OI,求调最小生成树
查看原帖
蒟蒻刚学1小时OI,求调最小生成树
740254
To_2051楼主2024/3/24 17:38

rt,玄关

#include<bits/stdc++.h>
using namespace std;
int ans,n,m,cnt;
struct node{
	int u,v,w;
}a[100001];
int fa[100001];
bool cmp(node a,node b)
{
	return a.w<b.w;
}
int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
bool work()
{
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int eu=find(a[i].u),ev=find(a[i].v);
		if(eu==ev) continue;
		cnt++;ans+=a[i].w;
		fa[eu]=ev;
		if(cnt==n) 
		{
			cout<<"orz";
			return 0;
		}
	}
	return 1;
}
int main()
{
	int pos;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		pos++;
		cin>>a[pos].u>>a[pos].v>>a[pos].w;
	}
	sort(a+1,a+m+1,cmp);
	if(work()) cout<<ans;
	return 0;
}
2024/3/24 17:38
加载中...