Rust用函数式方式从Vector构建HashMap并修改时报错如何解决

报错根因 #

• 你误用了迭代器的map算子：map设计用途是转换迭代器元素，要求闭包返回转换后的新值，不适合执行修改外部状态的副作用操作。你在map闭包中返回了entry()方法产生的&mut i32可变引用，这个引用指向外部捕获的words哈希表，违反了FnMut闭包的约束：不允许捕获变量的引用逃出闭包生命周期。
• 附加隐患：Rust迭代器是惰性执行的，如果你不主动消费map返回的迭代器（比如调用collect、count等方法），你写的词频统计逻辑根本不会执行。

修正方案 #

方案1：纯函数式fold实现（推荐） #

fold是函数式编程中专门用于迭代累加的算子，不需要外部可变状态，完全符合函数式写法需求：

use std::collections::HashMap;

pub fn can_construct_note(magazine: &[&str], note: &[&str]) -> bool {
    // 提前剪枝优化：笔记长度大于杂志直接返回false
    if note.len() > magazine.len() {
        return false;
    }

    // fold迭代统计杂志词频，所有状态都在闭包内部流转，无外部可变性泄露
    let mut word_counts = magazine.iter().fold(HashMap::new(), |mut counts, &word| {
        counts.entry(word).and_modify(|cnt| *cnt += 1).or_insert(1);
        counts
    });

    // all算子短路校验笔记是否可构造，遇到第一个缺失词直接返回false
    note.iter().all(|&word| {
        match word_counts.get_mut(word) {
            Some(cnt) if *cnt > 0 => {
                *cnt -= 1;
                true
            }
            _ => false
        }
    })
}

方案2：替换map为for_each #

如果要保留外部声明哈希表的写法，把map替换为专门用于执行副作用的for_each算子即可解决报错：

use std::collections::HashMap;

pub fn can_construct_note(magazine: &[&str], note: &[&str]) -> bool {
    if note.len() > magazine.len() {
        return false;
    }

    let mut words: HashMap<&str, i32> = HashMap::new();

    // for_each闭包无返回值，不会产生引用逃逸，且会立即消费迭代器执行逻辑
    magazine.iter().for_each(|&w| {
        words.entry(w).and_modify(|e| *e += 1).or_insert(1);
    });

    note.iter().all(|&w| {
        match words.get_mut(w) {
            Some(cnt) if *cnt > 0 => {
                *cnt -= 1;
                true
            }
            _ => false
        }
    })
}

内容的提问来源于stack exchange，提问作者luceat-lux-vestra