Plantilla simple de C ++ Kruskal

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int f[200005];
int n;
int m;
int ans=0;
int cnt=0;
struct edge{
    
    
	int u;
	int v;
	int w;
}Edge[200005];
int read()
{
    
    
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9')
	{
    
    
		if(c=='-')
		f=-1;
		c=getchar();
	} 
	while(c>='0'&&c<='9')
	{
    
    
		x=(x<<3)+(x<<1)+(c^48);
		c=getchar();
	}
	return x*f;
}
int find(int x)
{
    
    
	while(x!=f[x])
	{
    
    
		x=f[x];
	}
	return x;
}
int cmp(struct edge a,struct edge b){
    
    
	return a.w<b.w;
}
void kruskal(){
    
    
	for(int i=0;i<m;i++)
	{
    
    
		int eu,ev;
		eu=find(Edge[i].u);
		ev=find(Edge[i].v);
		if(eu==ev)
		continue;
		ans+=Edge[i].w;
		f[eu]=ev;
		cnt++;
		if(cnt==n)
		break; 
	}
}
int main(){
    
    

	n=read();
	m=read();
	for(int i=1;i<=n;i++)
	{
    
    
		f[i]=i;
	}
	for(int i=0;i<m;i++)
	{
    
    
		Edge[i].u=read();
		Edge[i].v=read();
		Edge[i].w=read();	
	}
	sort(Edge,Edge+m,cmp);
	kruskal();
	cout<<ans<<endl;
} 

Supongo que te gusta

Origin blog.csdn.net/qq_43663263/article/details/107162990
Recomendado
Clasificación