Rust

2025年10月30日更新 10 人订阅
原价: ¥ 6 限时优惠
专栏简介 Rust编程语言之错误处理 Rust 语言之 flod Rust编程语言之Cargo、Crates.io详解 Rust编程语言之枚举与模式匹配 Rust语言 - 接口设计的建议之受约束(Constrained) Rust编程语言之无畏并发 Rust语言 - 接口设计的建议之灵活(flexible) Rust语言 - 接口设计的建议之显而易见(Obvious) Rust语言 - 接口设计的建议之不意外(unsurprising) Rust 实战:构建实用的 CLI 工具 HTTPie Rust编程语言学习之高级特性 Rust内存管理揭秘:深度剖析指针与智能指针 解决Rust中数组和切片的编译时大小问题 《Rust编程之道》学习笔记一 Rust Async 异步编程 简易教程 使用 Async Rust 构建简单的 P2P 节点 Rust编程语言入门之模式匹配 Rust async 编程 Rust编程语言之编写自动化测试 Rust编程语言之函数式语言特性:迭代器和闭包 《Rust编程之道》学习笔记二 Rust Tips 比较数值 使用 Rust 开发一个微型游戏 Rust编程初探:深入理解Struct结构体 深入理解Rust中的内存管理:栈、堆与静态内存详解 深入理解 Rust 结构体:经典结构体、元组结构体和单元结构体的实现 深入掌握 Rust 结构体:从模板到实例化的完整指南 深入理解Rust中的结构体:逻辑与数据结合的实战示例 深入理解 Rust 枚举:从基础到实践 掌握Rust字符串的精髓:String与&str的最佳实践 全面解析 Rust 模块系统:实战案例与应用技巧 Rust 中的 HashMap 实战指南:理解与优化技巧 掌握Rust模式匹配:从基础语法到实际应用 Rust 中的面向对象编程:特性与实现指南 深入理解 Rust 的 Pin 和 Unpin:理论与实践解析 Rust Trait 与 Go Interface:从设计到实战的深度对比 从零开始:用 Rust 和 Axum 打造高效 Web 应用 Rust 错误处理详解:掌握 anyhow、thiserror 和 snafu Rust 如何优雅实现冒泡排序 链表倒数 K 节点怎么删?Python/Go/Rust 实战 用 Rust 玩转数据存储:JSON 文件持久化实战 Rust实战:打造高效字符串分割函数 如何高效学习一门技术:从知到行的飞轮效应 Rust 编程入门:Struct 让代码更优雅 Rust 编程:零基础入门高性能开发 用 Rust 写个猜数游戏,编程小白也能上手! Rust 入门教程:变量到数据类型,轻松掌握! 深入浅出 Rust:函数、控制流与所有权核心特性解析 从零开始:用 Rust 和 Axum 打造高效 Web 服务 Rust 集合类型解析:Vector、String、HashMap 深入浅出Rust:泛型、Trait与生命周期的硬核指南 Rust实战:博物馆门票限流系统设计与实现 用 Rust 打造高性能图片处理服务器:从零开始实现类似 Thumbor 的功能 Rust 编程入门实战:从零开始抓取网页并转换为 Markdown 深入浅出 Rust:高效处理二进制数据的 Bytes 与 BytesMut 实战 Rust智能指针:解锁内存管理的进阶之道 用 Rust 打造命令行利器:从零到一实现 mini-grep 解锁Rust代码组织:轻松掌握Package、Crate与Module Rust 所有权:从内存管理到生产力释放 深入解析 Rust 的面向对象编程:特性、实现与设计模式 Rust + Protobuf:从零打造高效键值存储项目 bacon 点燃 Rust:比 cargo-watch 更爽的开发体验 用 Rust 打造微型游戏:从零开始的 Flappy Dragon 开发之旅 函数式编程的Rust之旅:闭包与迭代器的深入解析与实践 探索Rust编程之道:从设计哲学到内存安全的学习笔记 精读《Rust编程之道》:吃透语言精要,彻底搞懂所有权与借用 Rust 避坑指南:搞定数值比较,别再让 0.1 + 0.2 != 0.3 困扰你! 告别 Vec!掌握 Rust bytes 库,解锁零拷贝的真正威力 告别竞态条件:基于 Axum 和 Serde 的 Rust 并发状态管理最佳实践 Rust 异步编程实践:从 Tokio 基础到阻塞任务处理模式 Rust 网络编程实战:用 Tokio 手写一个迷你 TCP 反向代理 (minginx) 保姆级教程:Zsh + Oh My Zsh 终极配置,让你的 Ubuntu 终端效率倍增 不止于后端:Rust 在 Web 开发中的崛起之路 (2024数据解读) Rust核心利器:枚举(Enum)与模式匹配(Match),告别空指针,写出优雅健壮的代码 Rust 错误处理终极指南:从 panic! 到 Result 的优雅之道 想用 Rust 开发游戏?这份超详细的入门教程请收好! 用 Rust 实现 HTTPie:一个现代 CLI 工具的构建过程 Rust 异步实战:从0到1,用 Tokio 打造一个高性能并发聊天室 深入 Rust 核心:彻底搞懂指针、引用与智能指针 Rust 生产级后端实战:用 Axum + sqlx 打造高性能短链接服务 深入 Rust 内存模型:栈、堆、所有权与底层原理 Rust 核心概念解析:引用、借用与内部可变性 掌握 Rust 核心:生命周期与借用检查全解析 Rust 内存布局深度解析:从对齐、填充到 repr 属性 Rust Trait 分派机制:静态与动态的抉择与权衡 Rust Thread::Builder 用法详解:线程命名与栈大小设置 Rust 泛型 Trait:关联类型与泛型参数的核心区别 Rust Scoped Threads 实战:更安全、更简洁的并发编程 Rust 核心设计:孤儿规则与代码一致性解析 Rust 实战:从零构建一个多线程 Web 服务器 Rust Web 开发实战:构建教师管理 API 硬核实战:从零到一,用 Rust 和 Axum 构建高性能聊天服务后端 Rust Web 开发实战:使用 SQLx 连接 PostgreSQL 数据库 硬核入门:从零开始,用 Actix Web 构建你的第一个 Rust REST API (推荐 🔥) Rust 并发编程:详解线程间数据共享的几种核心方法 Rust并发安全基石:Mutex与RwLock深度解析 Rust Web实战:构建优雅的 Actix Web 统一错误处理 煮咖啡里的大学问:用 Rust Async/Await 告诉你如何边烧水边磨豆 深入浅出:Rust 原子类型与多线程编程实践 Rust 并发编程利器:OnceCell 与 OnceLock 深度解析 Rust 懒人编程:LazyCell 与 LazyLock 的惰性哲学 Rust 入门精髓:详解 Enum 的三种魔法,从模式匹配到状态管理 Rust 字符串魔法:String 与 &str 的深度解析与实践 Rust 模块化编程:驾驭代码结构与可见性的三大法则 Rust 实用进阶:深度剖析 Rust 生命周期的奥秘 Rust 智能指针大揭秘:Box、Rc、Arc、Cow 深度剖析与应用实践 Rust 并发编程三步曲:Join、Arc<Mutex> 与 mpsc 通道同步实战 Rust 声明宏实战进阶:从基础定义到 #[macro_export] 与多规则重载 Rust 类型转换实战:利用 From/Into Trait 实现带 Default 容错的安全转换 Rust 实战:实现 FromStr Trait,定制化字符串 parse() 与精确错误报告 Rust 实战:TryFrom Trait——如何在类型转换中强制执行业务逻辑检查 Rust 泛型编程基石:AsRef 和 AsMut 的核心作用与实战应用 揭秘 Rust Unsafe 编程:程序员接管内存安全的契约与实践 Rust FFI 入门:extern、ABI 与底层符号链接解析 Rust性能优化:零内存拷贝的链表合并技术实战 Rust 进阶:用 NonNull 裸指针实现高性能双向链表 O(N) 反转实战 Rust实战:如何用泛型和特性实现一个高性能、通用的插入排序 Rust实战:深度解析二叉搜索树(BST)的实现,掌握泛型与内存安全 用 Rust 优雅实现图搜索核心算法:广度优先搜索 (BFS) 实战 Rust 多线程的高效等待术:park() 与 unpark() 信号通信实战 Rust 并发加速器:用 Condvar 实现线程间“精确握手”与高效等待 Rust 算法精讲:用 DFS 玩转图遍历,从起点“一走到底”的秘密 Rust 并发实战:用 MPSC 通道构建线程安全的“任务指挥中心” 一行代码提速 30 倍!Rust Rayon 并行计算:告别多线程管理困境 Rust 实战:使用自定义泛型栈实现高效、严谨的括号匹配算法 Rust 并行加速:4 个实操案例,深度解析 Rayon 线程池的 Fork-Join 与广播机制 Rust 实战:用两个队列实现栈——重温经典数据结构面试题 Rust Async/Await 实战:从串行到并发,掌握 block_on 与 join! 的异步魔力 Rust 异步编程基石:Tokio 运行时从入门到精通(单线程与多线程实战) 告别重复造轮子:用 Rust 实现一个可大可小的通用“万能”二叉堆!

告别重复造轮子:用 Rust 实现一个可大可小的通用“万能”二叉堆!

告别重复造轮子:用Rust实现一个可大可小的通用“万能”二叉堆!在高性能编程领域,堆(Heap)是一个基石般的数据结构,广泛应用于优先队列、调度算法、以及各种高效的排序场景中。Rust语言标准库虽然提供了std::collections::BinaryHeap,但它是一个最大堆。本篇文

告别重复造轮子:用 Rust 实现一个可大可小的通用“万能”二叉堆!

在高性能编程领域,堆(Heap) 是一个基石般的数据结构,广泛应用于优先队列、调度算法、以及各种高效的排序场景中。Rust 语言标准库虽然提供了 std::collections::BinaryHeap,但它是一个最大堆。

本篇文章将带你深入理解堆的原理,并利用 Rust 的泛型、函数指针和 Iterator Trait 的强大组合,亲手打造一个通用、灵活且高性能的二叉堆(Heap&lt;T>)。它能够通过一个简单的比较函数切换,在最小堆(MinHeap)最大堆(MaxHeap) 之间自由转换,让你彻底掌握这一核心算法结构。

实操

实现一个 通用的二叉堆(Binary Heap)

/*
    heap
    implement a binary heap function
*/

use std::cmp::Ord;
use std::default::Default;

pub struct Heap&lt;T>
where
    T: Default,
{
    count: usize,
    items: Vec&lt;T>,
    comparator: fn(&T, &T) -> bool,
}

impl&lt;T> Heap&lt;T>
where
    T: Default,
{
    pub fn new(comparator: fn(&T, &T) -> bool) -> Self {
        Self {
            count: 0,
            items: vec![T::default()],
            comparator,
        }
    }

    pub fn len(&self) -> usize {
        self.count
    }

    pub fn is_empty(&self) -> bool {
        self.len() == 0
    }

    pub fn add(&mut self, value: T) {
        self.count += 1;

        // 1. 将新值添加到 items 的末尾(即 items[self.count])
        if self.items.len() > self.count {
            self.items[self.count] = value;
        } else {
            self.items.push(value);
        }

        // 2. 向上浮动 (Sift-Up)
        let mut current_idx = self.count;

        // 只要当前节点不是根节点 (idx > 1)
        while current_idx > 1 {
            let parent_idx = self.parent_idx(current_idx);

            // 检查当前节点和父节点是否违反堆属性
            // (即在 MinHeap 中,子节点比父节点小;在 MaxHeap 中,子节点比父节点大)
            if (self.comparator)(&self.items[current_idx], &self.items[parent_idx]) {
                // 违反属性,交换
                self.items.swap(current_idx, parent_idx);
                current_idx = parent_idx;
            } else {
                // 属性已满足,停止上浮
                break;
            }
        }
    }

    fn parent_idx(&self, idx: usize) -> usize {
        idx / 2
    }

    fn children_present(&self, idx: usize) -> bool {
        self.left_child_idx(idx) &lt;= self.count
    }

    fn left_child_idx(&self, idx: usize) -> usize {
        idx * 2
    }

    fn right_child_idx(&self, idx: usize) -> usize {
        self.left_child_idx(idx) + 1
    }

    fn smallest_child_idx(&self, idx: usize) -> usize {
        let left_idx = self.left_child_idx(idx);
        let right_idx = self.right_child_idx(idx);

        // 1. 检查是否有右子节点
        if right_idx > self.count {
            // 只有左子节点(或没有子节点,但 children_present 已经保证了至少有左子节点)
            left_idx
        } else {
            // 2. 左右子节点都存在,使用比较器判断哪个更符合堆属性
            // (self.comparator)(a, b) 为 true,则 a 是我们想要的 (e.g., MinHeap 中 a 较小)
            if (self.comparator)(&self.items[left_idx], &self.items[right_idx]) {
                left_idx
            } else {
                right_idx
            }
        }
    }
}

impl&lt;T> Heap&lt;T>
where
    T: Default + Ord,
{
    /// Create a new MinHeap
    pub fn new_min() -> Self {
        Self::new(|a, b| a &lt; b)
    }

    /// Create a new MaxHeap
    pub fn new_max() -> Self {
        Self::new(|a, b| a > b)
    }
}

impl&lt;T> Iterator for Heap&lt;T>
where
    T: Default,
{
    type Item = T;

    fn next(&mut self) -> Option&lt;T> {
        if self.is_empty() {
            return None;
        }

        // 1. 交换根节点 (index 1) 和最后一个元素 (index self.count)
        self.items.swap(1, self.count);

        // 2. 弹出并返回旧的根节点(现在在末尾)
        self.count -= 1;
        // 因为 self.items[0] 是默认值,所以我们pop掉最后一个元素是安全的
        let extracted_value = self.items.pop().unwrap_or_default();

        // 3. 向下沉降 (Sift-Down)
        let mut current_idx = 1;

        while self.children_present(current_idx) {
            // 找到最符合堆属性的子节点索引
            let target_child_idx = self.smallest_child_idx(current_idx);

            // 检查当前节点是否违反堆属性与目标子节点进行比较
            // 如果子节点比当前节点更符合堆属性 (e.g., MinHeap 中子节点更小)
            if (self.comparator)(&self.items[target_child_idx], &self.items[current_idx]) {
                // 违反属性,交换
                self.items.swap(current_idx, target_child_idx);
                current_idx = target_child_idx;
            } else {
                // 属性已满足,停止下沉
                break;
            }
        }

        Some(extracted_value)
    }
}

pub struct MinHeap;

impl MinHeap {
    #[allow(clippy::new_ret_no_self)]
    pub fn new&lt;T>() -> Heap&lt;T>
    where
        T: Default + Ord,
    {
        Heap::new(|a, b| a &lt; b)
    }
}

pub struct MaxHeap;

impl MaxHeap {
    #[allow(clippy::new_ret_no_self)]
    pub fn new&lt;T>() -> Heap&lt;T>
    where
        T: Default + Ord,
    {
        Heap::new(|a, b| a > b)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_empty_heap() {
        let mut heap = MaxHeap::new::&lt;i32>();
        assert_eq!(heap.next(), None);
    }

    #[test]
    fn test_min_heap() {
        let mut heap = MinHeap::new();
        heap.add(4);
        heap.add(2);
        heap.add(9);
        heap.add(11);
        assert_eq!(heap.len(), 4);
        assert_eq!(heap.next(), Some(2));
        assert_eq!(heap.next(), Some(4));
        assert_eq!(heap.next(), Some(9));
        heap.add(1);
        assert_eq!(heap.next(), Some(1));
    }

    #[test]
    fn test_max_heap() {
        let mut heap = MaxHeap::new();
        heap.add(4);
        heap.add(2);
        heap.add(9);
        heap.add(11);
        assert_eq!(heap.len(), 4);
        assert_eq!(heap.next(), Some(11));
        assert_eq!(heap.next(), Some(9));
        assert_eq!(heap.next(), Some(4));
        heap.add(1);
        assert_eq!(heap.next(), Some(2));
    }
}

这段 Rust 代码完整实现了一个 通用的二叉堆(Binary Heap),支持同时构建 最小堆 (MinHeap)最大堆 (MaxHeap),并且通过泛型和函数指针实现了灵活的比较逻辑。

📘 Rust 二叉堆实现详解

这段代码实现了一个通用的 堆 (Heap) 数据结构,它支持通过传入不同的比较函数构造 最小堆最大堆。堆是一种完全二叉树,常用于 优先队列 (Priority Queue)排序算法 (Heap Sort) 等场景。

🧩 1. 结构体定义

pub struct Heap&lt;T>
where
    T: Default,
{
    count: usize,                    // 当前堆中元素数量
    items: Vec&lt;T>,                   // 存储堆元素的动态数组,下标从 1 开始
    comparator: fn(&T, &T) -> bool,  // 比较函数,用于控制堆的性质(大顶堆/小顶堆)
}
  • items[0] 被保留为默认值(不参与运算),这使得计算父子节点索引更简单:
    • 父节点:idx / 2
    • 左子节点:idx * 2
    • 右子节点:idx * 2 + 1
  • comparator 是一个函数指针,它决定了堆的排序规则。 例如:
    • 对于最小堆:|a, b| a &lt; b
    • 对于最大堆:|a, b| a > b

⚙️ 2. 构造与基本操作

pub fn new(comparator: fn(&T, &T) -> bool) -> Self { ... }
pub fn len(&self) -> usize { self.count }
pub fn is_empty(&self) -> bool { self.count == 0 }

这些函数提供了堆的基本管理功能。new 初始化堆并设置比较逻辑,lenis_empty 提供统计信息。

🔼 3. 元素插入:上浮(Sift-Up)

pub fn add(&mut self, value: T) { ... }

插入新元素的逻辑是堆的核心:

  1. 将新元素插入数组末尾

    self.items.push(value);
    self.count += 1;
  2. 向上比较并交换 从新插入的节点开始,不断和父节点比较。如果违反堆的性质(例如在最小堆中,子节点比父节点更小),就交换它们的位置。

    while current_idx > 1 {
     let parent_idx = self.parent_idx(current_idx);
     if (self.comparator)(&self.items[current_idx], &self.items[parent_idx]) {
         self.items.swap(current_idx, parent_idx);
         current_idx = parent_idx;
     } else {
         break;
     }
    }

这一过程确保了堆在插入新元素后仍然保持有序。

🔽 4. 元素移除:下沉(Sift-Down)

实现了 Iterator trait 后,可以使用 .next() 来弹出堆顶元素。

fn next(&mut self) -> Option&lt;T> {
    if self.is_empty() { return None; }

    // 1. 交换根节点和最后一个节点
    self.items.swap(1, self.count);

    // 2. 弹出堆顶
    let extracted_value = self.items.pop().unwrap_or_default();
    self.count -= 1;

    // 3. 从根开始下沉
    let mut current_idx = 1;
    while self.children_present(current_idx) {
        let target_child_idx = self.smallest_child_idx(current_idx);
        if (self.comparator)(&self.items[target_child_idx], &self.items[current_idx]) {
            self.items.swap(current_idx, target_child_idx);
            current_idx = target_child_idx;
        } else {
            break;
        }
    }

    Some(extracted_value)
}

next() 会不断返回当前堆顶元素,并在每次调用后重新维护堆的有序性。 这样我们可以像迭代器一样使用堆来进行排序(堆排序的核心机制)。

🧮 5. MinHeap 和 MaxHeap 的封装

为方便用户使用,代码提供了两个结构体包装器:

pub struct MinHeap;
pub struct MaxHeap;

它们的 new() 方法分别指定不同的比较逻辑:

pub fn new&lt;T>() -> Heap&lt;T>
where
    T: Default + Ord,
{
    Heap::new(|a, b| a &lt; b)  // 最小堆
}
pub fn new&lt;T>() -> Heap&lt;T>
where
    T: Default + Ord,
{
    Heap::new(|a, b| a > b)  // 最大堆
}

🧪 6. 单元测试

代码附带的测试验证了堆的基本功能:

  • 空堆测试
  • 最小堆的正确弹出顺序(从小到大)
  • 最大堆的正确弹出顺序(从大到小)
#[test]
fn test_min_heap() {
    let mut heap = MinHeap::new();
    heap.add(4);
    heap.add(2);
    heap.add(9);
    assert_eq!(heap.next(), Some(2));
}

🧠 一句话

这段代码展示了如何使用 Rust 泛型 + 函数指针 + Iterator trait 实现一个 灵活、安全、通用的二叉堆结构,支持最小堆与最大堆两种模式,并具备上浮、下沉、插入、弹出、迭代等核心功能。

🧠 总结

这段 Rust 代码以工程实践的角度,完美地实现了通用二叉堆这一核心数据结构。其核心设计思想在于:

  1. 高度泛化 (Generics):使用 Heap&lt;T> 泛型结构,并通过 T: DefaultT: Ord 约束,确保代码的通用性和安全性。
  2. 灵活控制 (Comparator):利用函数指针 comparator: fn(&T, &T) -> bool,将堆的排序规则(大顶堆或小顶堆)与核心逻辑分离,实现了一套代码,两种功能
  3. 遵循规范 (Iterator Trait):通过实现 Iterator Trait 的 next() 方法,将堆顶元素的提取操作(pop)转化为标准的迭代行为,极大地增强了代码的 Rust 风格和可用性。
  4. 核心算法:清晰地实现了 上浮(Sift-Up)add 方法中)和 下沉(Sift-Down)next 方法中)两大核心操作,确保了堆的 $\mathcal{O}(\log n)$ 时间复杂度特性。

总而言之,这个实现不仅是数据结构的学习范本,也是 Rust 语言高级特性(如 Trait、泛型、闭包与函数指针)的优秀实践案例。

参考

点赞 0
收藏 0
分享
本文参与登链社区写作激励计划 ,好文好收益,欢迎正在阅读的你也加入。

0 条评论

请先 登录 后评论