更簡潔HashMap的初始化

Question

我使用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

可以使用迭代器來模擬字典理解，例如

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];一樣使用。