leetcode简单题19 N.100 相同的树 rust描述

rust 复制代码
// [1,2,3] [1,2,3] true
// [1,2] [1,null,2] false
// [1,2,1] [1,1,2] false
use std::rc::Rc;
use std::cell::RefCell;
#[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
    }
  }
}
// 递归解法
pub fn is_same_tree(p: Option<Rc<RefCell<TreeNode>>>, q: Option<Rc<RefCell<TreeNode>>>) -> bool {
    match (p, q) {
        (Some(p_node), Some(q_node)) => {
            let p_borrow = p_node.borrow();
            let q_borrow = q_node.borrow();
            p_borrow.val == q_borrow.val &&
                is_same_tree(p_borrow.left.clone(), q_borrow.left.clone()) &&
                is_same_tree(p_borrow.right.clone(), q_borrow.right.clone())
        }
        (None, None) => true,
        _ => false,
    }
}
// 迭代解法

pub fn is_same_tree2(p: Option<Rc<RefCell<TreeNode>>>, q: Option<Rc<RefCell<TreeNode>>>) -> bool {
    let mut stack = vec![(p, q)];

    while let Some((p, q)) = stack.pop() {
        match (p, q) {
            (Some(p_node), Some(q_node)) => {
                let p_borrow = p_node.borrow();
                let q_borrow = q_node.borrow();
                if p_borrow.val != q_borrow.val {
                    return false;
                }
                stack.push((p_borrow.left.clone(), q_borrow.left.clone()));
                stack.push((p_borrow.right.clone(), q_borrow.right.clone()));
            }
            (None, None) => continue,
            _ => return false,
        }
    }

        true
    }

fn main() {
    //根据[1,2,3] [1,2,3] true编写测试用例
    let tree1 = Some(Rc::new(RefCell::new(TreeNode {
        val:1,
        left:Some(Rc::new(RefCell::new(TreeNode {
            val:2,
            left:None,
            right:None,
        }))),
        right:Some(Rc::new(RefCell::new(TreeNode {
            val:3,
            left:None,
            right:None,
        }))),
    })));
    let tree2 = Some(Rc::new(RefCell::new(TreeNode {
        val:1,
        left:Some(Rc::new(RefCell::new(TreeNode {
            val:2,
            left:None,
            right:None,
        }))),
        right:Some(Rc::new(RefCell::new(TreeNode {
            val:3,
            left:None,
            right:None,
        }))),
    })));
    assert_eq!(is_same_tree(tree1.clone(), tree2.clone()), true);
    assert_eq!(is_same_tree2(tree1.clone(), tree2.clone()), true);
}
相关推荐
啊我不会诶12 小时前
24ICPC成都站补题
数据结构·算法
2401_8414956414 小时前
【计算机视觉】基于数学形态学的保留边缘图像去噪
人工智能·python·算法·计算机视觉·图像去噪·数学形态学·边缘保留
十八岁讨厌编程14 小时前
【算法训练营Day30】动态规划part6
算法·动态规划
CoderYanger15 小时前
优选算法-双指针:2.复写零
java·后端·算法·leetcode·职场和发展
charlie11451419115 小时前
理解C++20的革命特性——协程支持2:编写简单的协程调度器
c++·学习·算法·设计模式·c++20·协程·调度器
hadage23315 小时前
--- 常见排序算法汇总 ---
算法·排序算法
Mrs.Gril16 小时前
目标检测:yolov7算法在RK3588上部署
算法·yolo·目标检测
WWZZ202517 小时前
ORB_SLAM2原理及代码解析:单应矩阵H、基础矩阵F求解
线性代数·算法·计算机视觉·机器人·slam·基础矩阵·单应矩阵
2401_8414956417 小时前
【计算机视觉】分水岭实现医学诊断
图像处理·人工智能·python·算法·计算机视觉·分水岭算法·医学ct图像分割
liulilittle18 小时前
网络编程基础算法剖析:从字节序转换到CIDR掩码计算
开发语言·网络·c++·算法·通信