Question

我有Vec个可选HashSet。如果插槽是None，我想分配一个HashSet并将其存储在Vec中。在这两种情况下，我都想添加到HashSet中。

我了解编译器错误在抱怨什么，但没有哪种语法会使问题消失。编译器建议的更改以及我尝试的任何其他语法也都不会编译。如何匹配Vec单元格，检查它是否为None，如果它为Some<HashSet>，则分配一个None；或者，如果确实存在，则访问HashSet并在这两种情况下都向集合中添加新的整数？

代码：

use std::collections::HashSet;
use std::collections::BTreeSet;
use std::iter::FromIterator;
use maplit;

pub struct Graph {
    outgoing_edges : Vec<Option<HashSet<usize>>>,
    incoming_edges : Vec<Option<HashSet<usize>>>
}

impl Graph {
    pub fn new(node_count : usize) -> Self {
        Graph {
            outgoing_edges : vec!(None; node_count),
            incoming_edges : vec!(None; node_count)
        }
    }

    /// Add a directional edge that starts at `from_node` and points to `to_node`.
    pub fn add_edge(&mut self, from_node : usize, to_node : usize) {
        match &self.outgoing_edges[from_node] {
            Some(mut set) => { set.insert(to_node); () },
            None => { self.outgoing_edges[from_node] = Some(hashset!{ to_node }); () }
        }
        match &self.incoming_edges[to_node] {
            Some(mut set) => { set.insert(from_node); () },
            None => { self.incoming_edges[to_node] = Some(hashset!{ from_node }); () }
        }
    }
}

错误：

（行号来自我的原始文件，而不是短代码段。错误消息来自于我添加与号之前要借用，如上所示，但该代码更改无效。）

error[E0507]: cannot move out of index of `std::vec::Vec<std::option::Option<std::collections::HashSet<usize>>>`
  --> src\graph\mod.rs:46:15
   |
46 |         match self.outgoing_edges[from_node] {
   |               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ help: consider borrowing here: `&self.outgoing_edges[from_node]`
47 |             Some(mut set) => { set.insert(to_node); () },
   |                  -------
   |                  |
   |                  data moved here
   |                  move occurs because `set` has type `std::collections::HashSet<usize>`, which does not implement the `Copy` trait

Answer 1

如果需要，可以使用方法Option::get_or_insert_with()来创建新的哈希集，而不是使用模式匹配，并返回对现有或新创建的哈希集的引用。完整的代码如下所示：

pub fn add_edge(&mut self, from_node: usize, to_node: usize) {
    self.outgoing_edges[from_node]
        .get_or_insert_with(HashSet::new)
        .insert(to_node);
    self.incoming_edges[to_node]
        .get_or_insert_with(HashSet::new)
        .insert(from_node);
}

Answer 2

这非常简单，您希望通过引用进行绑定：

/// Add a directional edge that starts at `from_node` and points to `to_node`.
pub fn add_edge(&mut self, from_node : usize, to_node : usize) {
    match self.outgoing_edges[from_node] {
        Some(ref mut set) => { set.insert(to_node); },
        None => { self.outgoing_edges[from_node] = Some(hashset!{ to_node }); }
    }
    match self.incoming_edges[to_node] {
        Some(ref mut set) => { set.insert(from_node); },
        None => { self.incoming_edges[to_node] = Some(hashset!{ from_node }); }
    }
}

或者使用称为binding mode的模式匹配人体工程学功能，您还可以执行以下操作：

/// Add a directional edge that starts at `from_node` and points to `to_node`.
pub fn add_edge(&mut self, from_node : usize, to_node : usize) {
    match &mut self.outgoing_edges[from_node] {
        Some(set) => { set.insert(to_node); },
        None => { self.outgoing_edges[from_node] = Some(hashset!{ to_node }); }
    }
    match &mut self.incoming_edges[to_node] {
        Some(set) => { set.insert(from_node); },
        None => { self.incoming_edges[to_node] = Some(hashset!{ from_node }); }
    }
}

如何匹配和修改Vec <Option <HashSet >>中的元素？

2 个答案: