web-dev-qa-db-ja.com

HashMapを効率的に検索および挿入する方法は？

私は次のことをしたいです：

  • 特定のキーの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

私は結局このようなことをすることになりましたが、それがルックアップを2回実行するという事実が好きではありません（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!");
    }
};

1つのmatchだけでこれを行う安全な方法はありますか？

hashmaprustlookup
78
Yusuke Shinyama

entry AP​​I はこのために設計されています。マニュアル形式では、次のようになります

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)
97
huon