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있습니까?

답변:


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)

신속한 답변 감사합니다! 이제 문서를 좀 더 자세히 살펴보아야한다는 것을 배웠습니다.
Yusuke Shinyama 2015

22
entry ()의 문제는 항상 키를 복제해야한다는 것입니다. 이것을 피할 수있는 방법이 있습니까?
Pascalius

당신은 당신의 키 유형을 만들 수 @Pascalius &T또는 (키지도, 예를 들어 정적 문자열 오래 살 경우) Rc<T>대신를 T-하지만 어느 경우에 꽤 아니다
kbolino

@Pascalius : v.key()에 대한 표현식에서 사용할 수 있습니다 . default그러면 해시 맵에있는 키에 대한 참조를 얻을 수 있으므로이 방법으로 복제를 피할 수 있습니다
Chris Beck
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.