如何有效地从HashMap查找和插入?


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_keymap.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

Answers:


119

entryAPI是专为这一点。以手动形式,可能看起来像

use std::collections::hash_map::Entry;

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

或者可以使用简要形式:

map.entry(key).or_insert_with(|| default)

如果default即使没有插入也可以进行计算/便宜,它也可以是:

map.entry(key).or_insert(default)

感谢您的快速分析!现在,我了解到我应该深入研究文档。
雄介

22
entry()的问题在于,您始终必须克隆密钥,有没有办法避免这种情况?
Pascalius

@Pascalius,您可以设置您的键类型&T(如果键的寿命超过了地图,例如静态字符串),或者Rc<T>代替T它-但这两种情况都不是很好
kbolino

@Pascalius:您可以v.key()在表达式中使用default,然后它将获得对哈希表中存在的键的引用,因此您可以通过这种方式避免克隆
Chris Beck
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.