diff options
author | Jack Sangdahl <[email protected]> | 2025-03-10 00:36:13 -0600 |
---|---|---|
committer | Jack Sangdahl <[email protected]> | 2025-03-10 00:36:13 -0600 |
commit | 7c60dee83b436e8e8a667bf9f191beced8de4fc3 (patch) | |
tree | eb7a24ee2067f92245c57bb118e85a1baf15df50 /other/leaf_similar_trees.rs | |
parent | 5d7cdc215e4ba1ec4cbb24c084cccb5e5e641468 (diff) | |
download | leetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.tar.gz leetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.zip |
chore: move cpp and rust code to directory
Diffstat (limited to 'other/leaf_similar_trees.rs')
-rw-r--r-- | other/leaf_similar_trees.rs | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/other/leaf_similar_trees.rs b/other/leaf_similar_trees.rs new file mode 100644 index 0000000..6dcaae1 --- /dev/null +++ b/other/leaf_similar_trees.rs @@ -0,0 +1,44 @@ +#[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 + } + } +} + +use std::rc::Rc; +use std::cell::RefCell; +impl Solution { + pub fn leaf_similar(root1: Option<Rc<RefCell<TreeNode>>>, root2: Option<Rc<RefCell<TreeNode>>>) -> bool { + let mut leaves1 = Vec::<i32>::new(); + Solution::get_leaves(root1, &mut leaves1); + let mut leaves2 = Vec::<i32>::new(); + Solution::get_leaves(root2, &mut leaves2); + leaves1 == leaves2 + } + fn get_leaves(node: Option<Rc<RefCell<TreeNode>>>, leaves: &mut Vec<i32>) { + if let Some(node) = node { + let node = node.borrow(); + if node.left == None && node.right == None { + leaves.push(node.val); + } else { + if let Some(ref left_node) = node.left { + Solution::get_leaves(Some(Rc::clone(left_node)), leaves); + } + if let Some(ref right_node) = node.right { + Solution::get_leaves(Some(Rc::clone(right_node)), leaves); + } + } + } + } +} |