克鲁斯卡尔算法模板(并查集完美结合)

#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 10005
using namespace std;
int f[N],rank[N];
struct node
{
	int x,y,val;
}e[N];
bool operator<(node a , node b)
{
	return a.val<b.val;
}
void init_set()
{
	for(int i=0 ; i<=N ; i++) 
		f[i]=i,rank[i]=0;
}
int find_set(int k)
{
	/*if(k!=f[k]) f[k]=find_set(f[k]);
	return f[k];*/
	return f[k]=k!=f[k]?find_set(f[k]):f[k];

}
void union_set(int x , int y)
{
	x=find_set(x);
	y=find_set(y);
	if(x==y) return ; 
	else
	{
		if(rank[x]>rank[y]) f[y]=x;
		else if(rank[y]>rank[x]) f[x]=y;
		else  rank[x]++,f[y]=x;
	}
}
int main()
{
	//freopen("1.txt","r",stdin);
	int n,i,j,s;
	while(scanf("%d",&n))
	{
		if(n==0) break;
		int a,b,c,d;
		for(i=0 ; i<n*(n-1)>>1 ; i++)
		{
			scanf("%d%d%d%d",&a,&b,&c,&d);
			if(d==1) c=0;
			e[i].x=a,e[i].y=b,e[i].val=c;
		}
		sort(e,e+(n*(n-1)>>1));
		init_set();
		for(s=j=0 ; j<n*(n-1)>>1 ; j++)
		{
			a=find_set(e[j].x);
			b=find_set(e[j].y);
			if(a!=b)
			{
				union_set(a,b);
				s+=e[j].val;
			}
		}
		printf("%d\n",s);
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值