rust - 更新 BTreeSet 中的结构

Question

我想修改 BTreeSet 中的结构。

我想实现类似以下的东西 here :

use std::collections::BTreeSet;

#[derive(Eq, PartialEq, Ord, PartialOrd, Debug)]
struct X {
    key: String,
    val: Option<i32>,
}

fn main() {
    let mut set: BTreeSet<X> = BTreeSet::new();
    set.insert(X {
        key: "a".to_string(),
        val: Some(1),
    });
    set.insert(X {
        key: "b".to_string(),
        val: Some(1),
    });
    
    nonify(&mut set, "a".to_string());
    println!("{:?}", set);
}

fn nonify(set: &mut BTreeSet<X>, k: String) {
    for mut s in set.iter() {
        if s.key == k {
            s.val = None;
        }
    }
}

这不起作用，因为 s 不是可变引用。

我如何完成这样的事情？

Answer 1

修改 BTreeSet 中的对象使得它们的顺序发生变化是不安全的(就程序逻辑而言)，因为这样做会破坏集合的内部结构 - 这是 relevant documentation 中的一个片段:

It is a logic error for an item to be modified in such a way that the item's ordering relative to any other item, as determined by the Ord trait, changes while it is in the set. This is normally only possible through Cell, RefCell, global state, I/O, or unsafe code.

我不知道您的确切用例，但由于您似乎使用的是键值对，因此 HashMap 或 BTreeMap 可能在这里起作用:出于同样的原因，它们不允许您改变键，但是与键关联的值可能会发生变化。