更简洁的HashMap初始化

Question

我用a HashMap来计算字符串中不同字符的出现次数:

let text = "GATTACA";
let mut counts: HashMap<char, i32> = HashMap::new();
counts.insert('A', 0);
counts.insert('C', 0);
counts.insert('G', 0);
counts.insert('T', 0);

for c in text.chars() {
    match counts.get_mut(&c) {
        Some(x) => *x += 1,
        None => (),
    }
}

是否有更简洁或声明的方式来初始化HashMap？例如在Python中我会这样做:

counts = { 'A': 0, 'C': 0, 'G': 0, 'T': 0 }

要么

counts = { key: 0 for key in 'ACGT' }

Answer 1

我在官方文档中看到的另一种方式：

use std::collections::HashMap;

fn main() {
    let timber_resources: HashMap<&str, i32> =
    [("Norway", 100),
     ("Denmark", 50),
     ("Iceland", 10)]
     .iter().cloned().collect();
    // use the values stored in map
}

Answer 2

您可以使用迭代器来模拟字典理解,例如

let counts = "ACGT".chars().map(|c| (c, 0_i32)).collect::<HashMap<_, _>>();

甚至 for c in "ACGT".chars() { counts.insert(c, 0) }.

此外,可以编写宏来允许任意值的简明初始化.

macro_rules! hashmap {
    ($( $key: expr => $val: expr ),*) => {{
         let mut map = ::std::collections::HashMap::new();
         $( map.insert($key, $val); )*
         map
    }}
}

用过像let counts = hashmap!['A' => 0, 'C' => 0, 'G' => 0, 'T' => 0];.

Answer 3

从 Rust 1.56 开始，您可以使用 键值对数组from()来构建 a 。Hashmap这使得可以简洁地初始化，而无需指定类型或编写宏。

use std::collections::HashMap;

fn main() {
    let m = HashMap::from([
        ('A', 0),
        ('C', 0),
        ('G', 0),
        ('T', 0)
    ]);
}