最小生成树——kruskal

codevs上的一道模板题,最小生成树的意思大概就是,给好多条边和权值,让用最小的代价(权值和)使这个图连通
如果这个图一共有n个点,那么做法就是对于每一个边的权值给边排序,从小到大,然后从小里选,如果选出的边的两个点已经在一个集合里了
就跳过,如果不在一个集合里,就并上这两点,同时ans加上这条边的权值,由于排过序,所以每次并边的代价一定是最小的,并且记录并边的
次数,如果次数==n-1,说明这个图就连通了,退出即可
#include<iostream>
#include<algorithm>
using namespace std;
struct line
{
	int from;
	int to;
	int w;
}edge[100001];
int map[101][101];
int n;
int tot;
int pre[101];
int rank[101];
void init()
{
	for(int i=1;i<=n;i++)
	{
		pre[i]=i;
		rank[i]=0;
	}
}
int find(int x)
{
	if(x==pre[x])return x;
	return pre[x]=find(pre[x]);
}
void unite(int x,int y)
{
    x=find(x);
    y=find(y);
	if(rank[x]>rank[y])pre[y]=x;
	else 
	{
		pre[x]=y;
		if(rank[x]==rank[y])rank[y]++;
	}
}
bool cmp(const line& a,const line& b)
{
	return a.w<b.w;
}
int main()
{
    std::ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			cin>>map[i][j];
			if(map[i][j]==0)continue;
			edge[++tot].from=i;
			edge[tot].to=j;
			edge[tot].w=map[i][j];
		}
	init();
	int ans=0;
	int k=0;
	sort(edge+1,edge+1+tot,cmp);
	for(int i=1;i<=tot;i++)
	{
		if(find(edge[i].from)!=find(edge[i].to))
		{
			unite(edge[i].from,edge[i].to);
			ans+=edge[i].w;
			k++;
		}
		if(k==n-1)break;
		
	}
	cout<<ans;
	return 0;
	
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值