Как эффективно искать и вставлять в HashMap?

103

Я бы хотел сделать следующее:

  • Найдите 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?

Юсуке Шиньяма
источник

Ответы:

121

Для этого разработан 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)
гуон
источник
Спасибо за быстрый ансер! Теперь я узнал, что должен заглянуть в документы немного глубже.
Юсуке Шиньяма,
23
проблема с entry () заключается в том, что вам всегда нужно клонировать ключ, есть ли способ избежать этого?
Паскалиус
@Pascalius, вы можете сделать свой тип ключа &T(если ключи переживают карту, например, статические строки) или Rc<T>вместо T- но это не красиво в любом случае
kbolino
@Pascalius: вы можете использовать v.key()в выражении для default, и тогда он получит ссылку на ключ в том виде, в котором он существует в хэш-карте, так что вы можете избежать клона таким образом,
Крис Бек,