HashMapを効率的に検索して挿入する方法は?


102

次のことを行いたいのですが。

  • Vec特定のキーのa を検索し、後で使用できるように保存します。
  • 存在しない場合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_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!");
    }
};

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

回答:


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、それが挿入されていない場合でも、計算するOK /安いです、それはまた、単にすることができます:

map.entry(key).or_insert(default)

迅速な回答をありがとう!これで、ドキュメントを少し深く調査する必要があることを学びました。
新山裕介2015

22
entry()の問題は、常にキーを複製する必要があることです。これを回避する方法はありますか?
パスカリウス2017

@Pascaliusあなたはあなたのキータイプを作ることができます&T(キーがマップより長生きしている場合、例えば静的文字列)またはRc<T>その代わりにT-しかしどちらの場合もきれいではありません
kbolino

@Pascalius:v.key()の式で使用できdefault、ハッシュマップに存在するキーへの参照を取得するため、この方法でクローンを回避できます
Chris Beck
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.