构建字符计数的 HashMap 的惯用 Rust 方法是什么？

Question

我想计算字符串中每个字母的出现次数。目标是建立一个HashMap<char,i32>键是字符串中的所有字符而值是出现次数的地方。

假设我正在遍历char字符串或输入文件中的值。对于 each char，如果还没有遇到过，我需要将它HashMap作为一个值为 1 的新键添加到 中，但是如果之前已经看到过，我需要增加该值。

这是有效的代码。请耐心等待，我对 Rust 很陌生：

use std::collections::HashMap;

fn main() {
    let mut letter_counts: HashMap<char,i32> = HashMap::new();

    let input_string = "Hello, world!";
    let char_vec: Vec<char> = input_string.to_lowercase().chars().collect();
    for c in char_vec {
        if let Some(x) = letter_counts.get_mut(&c) {
            *x = *x + 1;
        } else {
            letter_counts.insert(c,1);
        }
    }
    println!("{:?}",letter_counts);
}

我想知道的是，在 Rust 中是否有一种惯用的方法来做到这一点？通过惯用语，我的意思是有一个标准库类型（如 Python 的defaultdict），或 HashMap 上的一个方法（如 Java 的HashMap.computeIfAbsent），可以使这个更简单，更清晰，和/或比我所做的手动编码算法更不容易出错?

Answer 1

如果您使用该Entry界面，这对于您想要做的事情可能会更容易一些，并且可能会更惯用：

use std::collections::HashMap;

fn main() {
    let mut letter_counts: HashMap<char,i32> = HashMap::new();

    let input_string = "Hello, world!";
    let char_vec: Vec<char> = input_string.to_lowercase().chars().collect();
    for c in char_vec {
        *letter_counts.entry(c).or_insert(0) += 1;
    }
    println!("{:?}",letter_counts);
}

这使您可以在条目不存在时创建该条目并同时对其进行修改。

如果你想要一些更实用的东西，你可以这样做：

use std::collections::HashMap;

fn main() {
    let input_string = "Hello, world!";
    let letter_counts: HashMap<char, i32> =
        input_string
            .to_lowercase()
            .chars()
            .fold(HashMap::new(), |mut map, c| {
                *map.entry(c).or_insert(0) += 1;
                map
            });
    println!("{:?}", letter_counts);
}

这使用折叠来累积项目。

如果您正在寻找一个计算项目频率的标准库函数，那么没有。函数式方法足够优雅，我个人不认为这是一个错误，在这种情况下，我通常会使用这种方法来处理大多数普通的、惯用的 Rust 代码。使用迭代器在 Rust 中很常见。

正如其他人所提到的，对于某些情况，肯定有更专业的替代方法。