Create 0226-invert-binary-tree.rs

Gyanachand1 · Nov 19, 2023 · 08f9b44 · 08f9b44
1 parent 394cd47
commit 08f9b44
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/rust/0226-invert-binary-tree.rs b/rust/0226-invert-binary-tree.rs
@@ -0,0 +1,16 @@
+use std::rc::Rc;
+use std::cell::RefCell;
+impl Solution {
+ pub fn invert_tree(root: Option<Rc<RefCell<TreeNode>>>) -> Option<Rc<RefCell<TreeNode>>> {
+ root.map(|node| {
+ {
+ let mut node_ref = node.borrow_mut();
+ let left = node_ref.left.take();
+ let right = node_ref.right.take();
+ node_ref.right = Solution::invert_tree(left);
+ node_ref.left = Solution::invert_tree(right);
+ }
+ node
+ })
+ }
+}