จะค้นหาและแทรกลงใน HashMap อย่างมีประสิทธิภาพได้อย่างไร?


107

ฉันต้องการทำสิ่งต่อไปนี้:

  • ค้นหา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หรือไม่?

คำตอบ:


126

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

หรือสามารถใช้แบบฟอร์ม briefer:

map.entry(key).or_insert_with(|| default)

หากdefaultคำนวณได้ดี / ถูกแม้ว่าจะไม่ได้ใส่ก็สามารถ:

map.entry(key).or_insert(default)

ขอบคุณสำหรับ Anser ที่รวดเร็ว! ตอนนี้ฉันได้เรียนรู้แล้วว่าฉันควรมองลึกลงไปที่เอกสาร
Yusuke Shinyama

25
ปัญหาเกี่ยวกับรายการ () คือคุณต้องโคลนคีย์เสมอมีวิธีหลีกเลี่ยงสิ่งนี้หรือไม่?
Pascalius

@Pascalius คุณสามารถสร้างประเภทคีย์ของคุณได้&T(หากคีย์อยู่ได้นานกว่าแผนที่เช่นสตริงคงที่) หรือRc<T>แทนที่จะเป็นT- แต่มันก็ไม่สวยในทั้งสองกรณี
kbolino

@Pascalius: คุณสามารถใช้v.key()ในนิพจน์สำหรับdefaultแล้วมันจะได้รับการอ้างอิงถึงคีย์ตามที่มีอยู่ในแฮชแมปดังนั้นคุณอาจหลีกเลี่ยงการโคลนด้วยวิธีนี้
Chris Beck
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.