如何在不更新容器API的情况下向HashSet插入值并获取其引用？

这个问题在Rust里真的很典型——所有权规则虽然安全，但有时候确实会让我们在和集合打交道时有点卡壳。不过放心，不用修改HashSet的API也能完美实现你想要的功能！

核心思路：利用HashSet的entry API #

标准库的HashSet其实已经提供了一个非常适合这个场景的工具：entry方法。它可以帮我们一次性处理“元素存在”和“元素不存在”两种情况，而且只需要一次哈希查找，效率很高。

实现代码 #

首先，我们需要给函数加上必要的trait约束（毕竟HashSet要求元素必须实现Eq和Hash才能保证唯一性和查找效率），然后用entry API完成逻辑：

use std::collections::HashSet;
use std::hash::Hash;

fn insert_referring<A: Eq + Hash>(set: &mut HashSet<A>, element: A) -> &A {
    match set.entry(element) {
        std::collections::hash_set::Entry::Occupied(occupied_entry) => {
            // 元素已经存在，返回集合中已有元素的引用
            occupied_entry.into_mut()
        }
        std::collections::hash_set::Entry::Vacant(vacant_entry) => {
            // 元素不存在，插入后返回新元素的引用
            vacant_entry.insert()
        }
    }
}

为什么这个方案可行？ #

• 所有权处理：entry方法会接管传入的element所有权——如果元素不存在，就把它插入集合；如果已经存在，就直接丢弃传入的element（因为集合里已经有等价的元素了）。
• 引用返回：不管是已有的元素还是刚插入的元素，我们都能直接拿到它在集合中的引用，完全符合你想要的函数签名。
• 效率最优：整个流程只做一次哈希计算和查找，比“先插入再查找”的两次操作效率更高。

使用示例 #

来试试实际用一下这个函数：

fn main() {
    let mut words = HashSet::new();

    // 插入新元素，返回新元素的引用
    let hello_ref = insert_referring(&mut words, "hello".to_string());
    println!("第一次插入：{}", hello_ref); // 输出 "hello"

    // 插入重复元素，返回集合中已有元素的引用
    let hello_dup_ref = insert_referring(&mut words, "hello".to_string());
    println!("插入重复元素：{}", hello_dup_ref); // 输出 "hello"
    println!("两个引用指向同一个元素：{}", hello_ref as *const _ == hello_dup_ref as *const _); // 输出 true

    // 插入另一个新元素
    let world_ref = insert_referring(&mut words, "world".to_string());
    println!("插入新元素：{}", world_ref); // 输出 "world"
}

额外说明 #

• 函数的trait约束A: Eq + Hash是必须的，这和HashSet本身的要求一致，没有额外增加使用负担。
• 如果你需要返回可变引用，只需要把函数的返回类型改成&mut A即可，代码逻辑不需要大改。

内容的提问来源于stack exchange，提问作者Nikita Volkov