如何在Rust中从HashMap获取作为值的HashSet？

错误根源分析 #

你遇到的编译错误核心原因是：HashMap::values()返回的是指向HashSet的引用迭代器（即&HashSet<A>），而你试图直接将这类引用收集为HashSet<A>。标准库并未为HashSet<A>实现FromIterator<&HashSet<A>>——因为这个操作逻辑不明确：是要把每个HashSet<A>作为元素存入新HashSet？还是要合并所有HashSet里的元素？

以下分两种场景给出解决方案：

场景1：合并所有HashSet的元素到一个大HashSet #

如果目标是把HashMap中所有HashSet内的A实例合并成一个新的HashSet<A>，需要扁平化迭代并克隆元素：

#[derive(PartialEq, Eq, Hash, Clone)] // 必须为A实现Clone trait，用于复制元素所有权
struct A {
    name: String,
}
 
use std::collections::{HashMap, HashSet};

fn main() {
    let mut x: HashMap<i32, HashSet<A>> = HashMap::new();
    // 示例：给HashMap添加测试数据
    x.insert(1, HashSet::from_iter([A { name: "a".to_string() }, A { name: "b".to_string() }]));
    x.insert(2, HashSet::from_iter([A { name: "b".to_string() }, A { name: "c".to_string() }]));

    // 合并所有HashSet的元素到y中
    let y: HashSet<A> = x.values()
        .flat_map(|hs| hs.iter().cloned()) // 扁平化遍历每个HashSet的元素，克隆获取所有权
        .collect();

    println!("{:?}", y); // 输出: {A { name: "a" }, A { name: "b" }, A { name: "c" }}
}

如果不想克隆元素，可以改用引用类型的HashSet（注意生命周期约束）：

// 获取所有元素的引用，无需Clone
let y: HashSet<&A> = x.values()
    .flat_map(|hs| hs.iter())
    .collect();

场景2：获取特定键对应的HashSet #

如果只是想获取HashMap中某个键对应的HashSet，有两种方式：

获取引用（不转移所有权） #

用get方法获取指向HashSet的不可变引用，返回Option<&HashSet<A>>：

fn main() {
    let x: HashMap<i32, HashSet<A>> = HashMap::new();
    if let Some(hs_ref) = x.get(&1) {
        // 使用引用遍历或操作元素
        for a in hs_ref {
            println!("{}", a.name);
        }
    }
}

获取所有权（移除键值对） #

用remove方法从HashMap中移除该键值对，获取HashSet的所有权，返回Option<HashSet<A>>：

fn main() {
    let mut x: HashMap<i32, HashSet<A>> = HashMap::new();
    if let Some(hs_owned) = x.remove(&1) {
        // 完全拥有该HashSet，可随意修改
        hs_owned.insert(A { name: "new_item".to_string() });
    }
}

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