如何在Rust中实现HashMap上带有突变的嵌套循环

Question

我有以下（精简）Rust代码：

use std::collections::HashMap;
struct Node {
    weight:   f64,
    outbound: f64,
}
struct Graph {
    edges: HashMap<u32, HashMap<u32, f64>>,
    nodes: HashMap<u32, Node>,
}
impl Graph {
    fn mutate(&mut self) {
        for (key, value) in self.nodes.iter() {
            if self.edges.contains_key(key) {
                for (target, weight) in self.edges[key].iter() {
                    self.nodes.entry(*target).or_insert(Node::new()).weight;
                }
            }
        }
    }
}

然而，由于Rust所有权规则（游乐场），我无法编译代码：

graph.rs:88:25: 88:35 error: cannot borrow `self.nodes` as mutable because it is also borrowed as immutable [E0502]
graph.rs:88                         self.nodes.entry(*target).or_insert(Node::new()).weight;
                                    ^~~~~~~~~~

如果我将第一个循环改为使用HashMap::iter_mut()，我会得到一个不同的错误（操场）：

graph.rs:88:25: 88:35 error: cannot borrow `self.nodes` as mutable more than once at a time [E0499]
graph.rs:88                         self.nodes.entry(*target).or_insert(Node::new()).weight;
                                    ^~~~~~~~~~

如何在Rust中实现这种带有突变的嵌套循环？

Answer 1

在对数据结构进行迭代时，不能在数据结构中插入或删除元素。

据我所知，Rust迭代器也不支持修改（比如Java迭代器的remove()）。

因此，您只能选择以下选项：

• 如果只有少量修改，则可以收集它们并在迭代完成后执行。

• 如果大部分数据结构都被修改了，或者它足够小，复制的开销无关紧要，那么您可以创建一个新的、修改过的数据结构，在迭代后替换原始数据结构。这通常是惯用的解决方案，在像map、flat_map或filter这样的迭代器上使用高阶函数。