将BST的子节点替换为Rust中的孙节点

Question

我正在尝试实现节点的移除。当节点没有子节点时，我只需修改父节点的左侧或右侧。当节点有左子树或右子树时，如何删除节点并保留其子树？

这里是树节点实现：

use std::rc::Rc;
use std::cell::RefCell;
// Definition for a binary tree node.
#[derive(Debug, PartialEq, Eq)]
pub struct TreeNode {
pub val: i32,
pub left: Option<Rc<RefCell<TreeNode>>>,
pub right: Option<Rc<RefCell<TreeNode>>>,
}
impl TreeNode {
#[inline]
pub fn new(val: i32) -> Self {
TreeNode {
val,
left: None,
right: None
}
}
}

实现已给出，不能更改。

Answer 1

我自己想好了：

let right_node = node.as_mut().unwrap().borrow_mut().right.take().unwrap(); // It may be the left child.
node.replace(right_node);