浙江大学机试--继续畅通工程

本文通过解决村庄间修路问题,详细介绍并实现Kruskal算法,一种寻找加权图最小生成树的方法。利用并查集处理图中边的连接状态,确保生成树的正确构建。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目:牛客网链接

概述: 给定n个村庄,然后给出他们之间的路的长度和代价,和是否修好,问如何修路才能让代价最小,输出最小代价。

思路: 本题采用最小生成树 Kruskal算法(并查集)。相关的知识可以百度一下。注释中写思路。

#include <iostream>
#include <algorithm>
using namespace std;

//用于并查集中使用,存放父节点 
int father[2000];

struct Edge{
	int a, b, status, cost;
};

bool cmp(Edge a, Edge b)
{
	if(a.status == b.status)
	{
		return a.cost < b.cost;
	}
	else
	{
		return a.status > b.status;
	}
}
//查找父节点 
int find_father(int x)
{
	while(x != father[x])
	{
		x = father[x];
	}
	return x;
} 


int main()
{
	int n;
	Edge edge[50000];
	
	while(scanf("%d", &n) != EOF && n != 0)
	{
		for(int i = 1; i <= n * (n - 1) / 2; i++)
		{
			scanf("%d%d%d%d",&edge[i].a,&edge[i].b,&edge[i].cost,&edge[i].status);
		} 
		sort(edge + 1, edge + 1 + n * (n -1) / 2, cmp);
		int ans = 0;
		for(int i = 1 ; i <= n ; i++)father[i] = i;
		for(int i = 1; i <= n * (n - 1) / 2; i++)
		{
			
			int a = find_father(edge[i].a);
			int b = find_father(edge[i].b);
			if( a != b)
			{
				father[a] = b;
				
				if(edge[i].status == 0) 
				{ans += edge[i].cost; 
				//cout << "链接了" << edge[i].a  <<" 和" << edge[i].b  <<"花费"<< edge[i].cost<< endl;
				} //只有状态为0才参与计算
			}	
		}
		printf("%d\n",ans);
		
	}
	
	
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值