如何有效地从 HashMap 查找和插入 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

我最终做了类似的事情，但是我不喜欢它执行两次查找(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就可以做到这一点？

43651

小开

最佳答案

entryAPI就是为此而设计的。在手动形式下，它看起来像

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，或者即使没有插入 default也可以/廉价地进行计算，那么可以使用 Entry::or_insert:

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

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

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

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()), }; } }