【kruscal求最小生成树】继续畅通工程

大致思路:

Kruscal求最小生成树用到的知识:图的边的结构体、按边权排序、并查集(不产生回路)

但是这道题仍然需要精心思考,原因是有一些边是“已经存在的”,那么在计算最小生成树成本的时候应该只考虑未修成的边

要注意两个易错点: ① 多组数据,并查集father数组要初始化,一些全局数组要初始化。 ② 由于有些边已经存在,要先对已存在的边进行并查集的处理,否则在边权排序的时候它排到后面去了,你在生成最小生成树的时候就不晓得还以为没有回路其实有!

注:有可能没有最小生成树,判断方法:看有多少个并查集(father[i]==i),有多个则说明没有建成最小生成树。

AC代码:

#include<iostream>
#include<bits/stdc++.h>
using namespace std;

//就是求最小生成树
struct edge
{
	int u,v,w;
}e[1005];

bool cmp(edge a,edge b)
{
	if(a.w<b.w)
		return true;
	else 
		return false;
}
int cnt=1;
void insert(int a,int b,int w)
{
	e[cnt].u=a;
	e[cnt].v=b;
	e[cnt].w=w;
	cnt++;
}
//void insert2(int a,int b,int w)
//{
//	insert(a,b,w);
//	insert(b,a,w);
//}

int isBuilt[1005][1005];

int father[1005];
int get(int a)
{
	if(father[a]==a)
		return a;
	return get(father[a]);
}
void add(int a,int b)
{
	a = get(a);
	b = get(b);
	if(a!=b)
		father[a]=b;
}

int main()
{
	int N;
	while(scanf("%d",&N)!=EOF)
	{
		if(N==0)
			return 0;
			
		//完全不同的一个案例,重新初始化全局数组 
		for(int i=0;i<N+1;i++)
			father[i]=i; 	
		cnt=1; 
		
		for(int i=0;i<N;i++)
			for(int j=0;j<N;j++)
				isBuilt[i][j]=0;
		 
			
		for(int i=0;i<N*(N-1)/2;i++)
		{
			int a,b,w,c;
			cin>>a>>b>>w>>c;
			insert(a,b,w);
			isBuilt[a][b]=c;
			isBuilt[b][a]=c;
			if(c==1) //已经修好的,要把并查集给做好
				add(a,b); 
		}
		cnt--;
		int count=0; //最小生成树中的边的条数 
		int ans=0;//需要花多少钱 
		sort(e+1,e+1+cnt,cmp); //按边权从小到大排列边
		
		for(int i=1;i<=cnt && count<N-1;i++)
		{
			int a=e[i].u;
			int b=e[i].v;
			//检查和之前加入的边是否产生回路 
			if(get(a)==get(b))
				continue;
			//否则,加入最小生成树
			add(a,b);
			count++;
			//但是要注意 如果已经修建好了就不用算钱了
			if(isBuilt[a][b]==0 || isBuilt[b][a]==0)
				ans+=e[i].w;
		} 
		cout<<ans<<endl;
	}
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值