公路村村通 -并查集

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node
{
    int u,v,len;
}a[3005];
int cmp(node x,node y)
{
    return x.len<y.len;
}
int f[1005];
int find(int x)
{
    if(x!=f[x])
        f[x]=find(f[x]);
    return f[x];
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].len);
    sort(a,a+m,cmp);
    for(int i=0;i<n;i++)
        f[i]=i;
    int ans=0;
    for(int i=0;i<m;i++)
    {
        int fa=find(a[i].u);
        int fb=find(a[i].v);
        if(fa!=fb)
        {
            if(fa>fb)
                f[fa]=fb;
            else
                f[fb]=fa;
            n--;
            ans+=a[i].len;
        }
    }
    if(n==1) printf("%d\n",ans);
    else printf("-1\n");
}
### 使用Kruskal算法解决公路村村通问题 #### 背景介绍 公路村村通问题是典型的最小生树(Minimum Spanning Tree, MST)问题之一。假设存在一组村庄以及连接这些村庄的道路本矩阵,目标是以最低的本修建道路使得所有村庄之能够互相连通。 Kruskal算法是一种经典的用于求解MST的方法[^1]。它的核心思想是从全局角度出发,在不形环的情况下逐步选择权值最小的边加入生树中,最终得到一棵总权重最小的生树[^2]。 --- #### Kruskal算法的具体应用过程 以下是利用Kruskal算法解决公路村村通问题的关键步骤: 1. **输入数据建模** 假设共有 \( n \) 个村庄和 \( m \) 条可能的道路,每条道路具有一定的建设费用作为权重。可以将此问题抽象一个无向加权图 \( G(V, E) \),其中顶点集合 \( V \) 示村庄,边集合 \( E \) 和对应的权重示各村庄之的潜在道路及其建设本。 2. **初始化并查集 (Union-Find)** 并查集被用来高效检测新选入的边是否会构回路。初始状态下,每个节点都属于独立的一个子集。每次功添加一条边时,需执行 `union` 操作来合并两个不同的子集[^3]。 3. **按权重排序所有边** 首先把所有的候选路径依据它们各自的造价从小到大排列好顺序以便后续逐一挑选最优选项。 4. **迭代选取合适的边构建MST** 对于排好序后的每一边依次考察: - 如果这条边两端别位于不同集合里,则将其纳入结果集中,并调用一次 union 合并操作; - 若发现两头已经处于同一组内则跳过该边以免造循环结构。 5. **终止条件** 当所选出的有效边数达到 \(|V|-1\) 即停止运算流程,此时所得即为满足题目需求的最佳解决方案——也就是所需铺设的所有路段组合起来形的那棵最小代价生树。 --- #### Python实现代码示例 下面给出一段基于上述原理编写而的Python程序片段演示如何运用克鲁斯卡尔方法处理此类实际场景中的最优化决策任务: ```python class UnionFind: def __init__(self, num_nodes): self.parent = list(range(num_nodes)) def find(self, u): if self.parent[u]!=u: self.parent[u]=self.find(self.parent[u]) return self.parent[u] def union(self,u,v): root_u=self.find(u) root_v=self.find(v) if root_u !=root_v : self.parent[root_u ]=root_v def kruskal(edges,nodes_count ): edges.sort(key=lambda edge :edge[2]) uf_set=UnionFind(nodes_count ) result_edges=[] for e in edges: node_a ,node_b,cost=e[:3] set_a=uf_set .find(node_a ) set_b=uf_set .find(node_b ) if(set_a!=set_b ): result_edges.append((node_a,node_b )) uf_set.union(set_a,set_b ) if(len(result_edges)==nodes_count-1 ):break total_cost=sum([e[-1]for e in result_edges ]) return {"edges":result_edges,"total_cost":total_cost} # Example usage with sample data representing villages and roads. villages_and_roads=[ (&#39;A&#39;,&#39;B&#39;,7),(&#39;A&#39;,&#39;D&#39;,5), (&#39;B&#39;,&#39;C&#39;,8),(&#39;B&#39;,&#39;D&#39;,9), (&#39;C&#39;,&#39;E&#39;,5),(&#39;C&#39;,&#39;F&#39;,6), (&#39;D&#39;,&#39;E&#39;,15),(&#39;D&#39;,&#39;G&#39;,6), (&#39;E&#39;,&#39;F&#39;,8),(&#39;E&#39;,&#39;H&#39;,9), (&#39;F&#39;,&#39;I&#39;,11),(&#39;G&#39;,&#39;H&#39;,10)] num_of_villages=len(set(sum(([list(pair[:-1])for pair in villages_and_roads]),[]))) solution=kruskal(villages_and_roads,num_of_villages) print(f"Selected Roads:{solution[&#39;edges&#39;]}\nTotal Cost={solution[&#39;total_cost&#39;]}") ``` --- #### 结果解释 运行以上脚本后会输出哪些具体线路应该建造才能达既定目的同时还保持整体花费最少的效果展示出来的同时也会告知总的预算数额是多少。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值