首页 文章

在具有可变引用的临时元组上进行模式匹配

提问于
浏览
2

我是Rust的新手,想知道是否有一种自然的方式来编写以下内容,同时仍然在元组上进行模式匹配 .

我有一个树状结构,有 keyvalueleftright 成员 . 我想创建一个看起来类似的插入函数

fn insert(&mut self, k: K, v: V) -> Option<V> {
    match (k.cmp(self.k.borrow()), &mut self.left, &mut self.right) {
        (Ordering::Less, &mut None, _) => { self.left = Some(Box::new(TreeNode::new(k, v))); None },
        (Ordering::Less, &mut Some(ref mut left), _) => left.insert(k, v),
        (Ordering::Greater, &mut None, _) => { self.right = Some(Box::new(TreeNode::new(k, v))); None },
        (Ordering::Greater, &mut Some(ref mut right), _) => right.insert(k, v),
        (Ordering::Equal, _, _) => { let mut temp = v; std::mem::swap(self.v.borrow_mut(), temp.borrow_mut()); Some(temp) },
    }
}

我喜欢这清楚地涵盖所有情况和结果操作,但是这不会编译因为 self.leftself.right 被match语句中使用的临时元组可变地借用,这意味着我无法在动作组中修改 self.leftself.right 这场比赛 . 以下替代方案有效

fn insert(&mut self, k: K, v: V) -> Option<V> {
    match k.cmp(self.k.borrow()) {
        Ordering::Less => match self.left {
            None => { self.left = Some(Box::new(TreeNode::new(k, v))); None },
            Some(ref mut left) => left.insert(k, v),
        },
        Ordering::Greater => match self.right {
            None => { self.right = Some(Box::new(TreeNode::new(k, v))); None },
            Some(ref mut right) => right.insert(k, v),
        },
        Ordering::Equal => { let mut temp = v; std::mem::swap(self.v.borrow_mut(), temp.borrow_mut()); Some(temp) }
    }
}

并且应该在逻辑上等同,但我觉得前者更清楚,更简洁地描述了问题 . 作为Rust的新手,我不确定是否有能力从元组中提取可变引用并将其从动作中的 None 更改为 Some 值,因此我想如果有任何替代方法我会忽略它 .

我已经将workingnon-working变体的短片段上传到了围栏 .

1 回答

  • 6

    有一种获取可变引用的方法,因为可以使用 @ -pattern(在Haskell中称为as-pattern)将标识符绑定到整个模式表示的值:

    fn insert(&mut self, k: K, v: V) -> Option<V> {
        match (k.cmp(self.k.borrow()), &mut self.left, &mut self.right) {
            (Ordering::Less, left @ &mut None, _) => {
                *left = Some(Box::new(TreeNode::new(k, v)));
                None
            }
        // ...
    

    这就像在匹配臂中编写 let left = &mut self.left; 一样,但它具有工作的有用优势 .

相关问题