rust 如何高效地从HashMap中查找和插入？

0yycz8jy 于 2024-01-08 发布在其他

关注(0)|答案(3)|浏览(102)

我想做以下事情：

为某个密钥创建Vec，并将其存储起来以供以后使用。
如果它不存在，则为键创建一个空的Vec，但仍将其保留在变量中。

如何有效地做到这一点？当然，我认为我可以使用match：

use std::collections::HashMap;

// This code doesn't compile.
let mut map = HashMap::new();
let key = "foo";
let values: &Vec<isize> = match map.get(key) {
    Some(v) => v,
    None => {
        let default: Vec<isize> = Vec::new();
        map.insert(key, default);
        &default
    }
};

字符串
当我尝试它时，它给了我这样的错误：

error[E0502]: cannot borrow `map` as mutable because it is also borrowed as immutable
  --> src/main.rs:11:13
   |
7  |     let values: &Vec<isize> = match map.get(key) {
   |                                     --- immutable borrow occurs here
...
11 |             map.insert(key, default);
   |             ^^^ mutable borrow occurs here
...
15 | }
   | - immutable borrow ends here

型
我最终做了这样的事情，但我不喜欢它执行两次查找（map.contains_key和map.get）的事实：

// This code does compile.
let mut map = HashMap::new();
let key = "foo";
if !map.contains_key(key) {
    let default: Vec<isize> = Vec::new();
    map.insert(key, default);
}
let values: &Vec<isize> = match map.get(key) {
    Some(v) => v,
    None => {
        panic!("impossiburu!");
    }
};

型
有没有一种安全的方法可以只用一个match来实现这一点？

rust

来源：https://stackoverflow.com/questions/28512394/how-to-lookup-from-and-insert-into-a-hashmap-efficiently

3条答案

按热度按时间

goucqfw61#

entry API就是为此而设计的。

let values = match map.entry(key) {
    Entry::Occupied(o) => o.into_mut(),
    Entry::Vacant(v) => v.insert(default),
};

字符串
可以通过Entry::or_insert_with使用更简短的形式：

let values = map.entry(key).or_insert_with(|| default);

型
如果default已经计算过了，或者即使没有插入也可以/便宜地计算，那么可以使用Entry::or_insert：

let values = map.entry(key).or_insert(default);

型
如果HashMap的值实现了Default，则可以使用Entry::or_default，尽管可能需要提供一些类型提示：

let values = map.entry(key).or_default();

型

赞(0）回复(0）举报 2024-01-08

nimxete22#

我使用了huon's answer并将其实现为trait：

use std::collections::HashMap;
use std::hash::Hash;

pub trait InsertOrGet<K: Eq + Hash, V: Default> {
    fn insert_or_get(&mut self, item: K) -> &mut V;
}

impl<K: Eq + Hash, V: Default> InsertOrGet<K, V> for HashMap<K, V> {
    fn insert_or_get(&mut self, item: K) -> &mut V {
        return match self.entry(item) {
            std::collections::hash_map::Entry::Occupied(o) => o.into_mut(),
            std::collections::hash_map::Entry::Vacant(v) => v.insert(V::default()),
        };
    }
}

字符串
然后我可以做：

use crate::utils::hashmap::InsertOrGet;

let new_or_existing_value: &mut ValueType = my_map.insert_or_get(my_key.clone());

型

赞(0）回复(0）举报 2024-01-08

6pp0gazn3#

只是为了在@huon的优秀答案中提供Entry解决方案的替代方案，您可以使用相对较新的try_insert方法分两步完成此操作，该方法仅在键尚未被占用时插入值。

let _ = map.try_insert(key, Vec::new());  // May or may not insert the empty vector.
let values = map.get_mut(key).unwrap();   // Guaranteed to be occupied now.

字符串
(Full Playground example here .）
到目前为止，try_insert方法仍然只在 nightly Rust中可用，在map_try_insert功能门之后。
具有讽刺意味的是，它目前是使用Entry API本身实现的。尽管在OP的情况下，我可能更喜欢直接使用entry和or_default，就像我自己接受的答案一样。

赞(0）回复(0）举报 2024-01-08

我来回答

rust 如何高效地从HashMap中查找和插入？

3条答案

相关问题

热门标签

最新问答