leetcode101. 对称二叉树,递归法+迭代法详解附代码

leetcode101. 对称二叉树

给你一个二叉树的根节点 root , 检查它是否轴对称。

示例 1:

输入:root = 1,2,2,3,4,4,3

输出:true

示例 2:

输入:root = 1,2,2,null,3,null,3

输出:false

递归法

1.定义辅助函数 check:这个函数接收两个参数,分别是指向二叉树节点的指针 p 和 q。它的作用是递归地检查从 p 和 q 出发的子树是否镜像对称。

2.处理空节点情况:在 check 函数中,首先检查 p 和 q 是否都为空。如果都为空,则说明这两个子树是对称的,返回 true。如果其中一个为空而另一个不为空,则说明这两个子树不对称,返回 false。

3.比较节点值:如果 p 和 q 都不为空,接下来比较它们的值。如果值不相等,则说明这两个子树不对称,返回 false。

4.递归检查子树:如果 p 和 q 的值相等,算法会继续递归地检查 p 的左子树和 q 的右子树是否对称,以及 p 的右子树和 q 的左子树是否对称。这一步骤是关键,因为它确保了对称性的检查是全面的。只有当这两个递归调用都返回 true 时,check 函数才会返回 true。

5.实现主函数 isSymmetric:这个函数是对外的接口,它接收一个参数 root,即二叉树的根节点。isSymmetric 函数通过调用 check 函数并传入 root 和 root 作为参数来实现对整个二叉树的对称性检查。这意味着算法会从根节点出发,检查其左右子树是否镜像对称。

递归法具体代码如下:

cpp 复制代码
class Solution {
public:
    bool check(TreeNode *p, TreeNode *q) {
        if (!p && !q) return true;
        if (!p || !q) return false;
        return p->val == q->val && check(p->left, q->right) && check(p->right, q->left);
    }

    bool isSymmetric(TreeNode* root) {
        return check(root, root);
    }
};

迭代法

递归代码变成迭代,往往需要辅助队列或者辅助栈,我们这里使用辅助队列。

初始节点入队列:首先把root节点两次入辅助队列。

层序遍历:使用一个 while 循环进行层序遍历,直到队列为空。在每次循环中,从队列中取出两个节点 u 和 v。

处理空节点:如果 u 和 v 都为空,说明当前层的两个节点是对称的,继续处理下一层。如果其中一个为空而另一个不为空,或者它们的值不相等,说明这两个子树不对称,函数返回 false。

对称性检查:如果 u 和 v 的值相等,算法会将 u 的左子节点和 v 的右子节点,以及 u 的右子节点和 v 的左子节点入队。这样,下一次循环时,它们会被取出并进行比较,确保左子节点与右子节点的值相等,从而保证对称性。

迭代法完整代码如下:

cpp 复制代码
class Solution {
public:
    bool check(TreeNode *u, TreeNode *v) {
        queue <TreeNode*> q;
        q.push(u); q.push(v);
        while (!q.empty()) {
            u = q.front(); q.pop();
            v = q.front(); q.pop();
            if (!u && !v) continue;
            if ((!u || !v) || (u->val != v->val)) return false;

            q.push(u->left); 
            q.push(v->right);

            q.push(u->right); 
            q.push(v->left);
        }
        return true;
    }

    bool isSymmetric(TreeNode* root) {
        return check(root, root);
    }
};
相关推荐
mCell2 小时前
Lua 编程入门:从基础语法到元表
javascript·算法·lua
wuyk5554 小时前
98.C语言易混难点:字符数组与字符串指针的底层差异
c语言·开发语言·c++·stm32·嵌入式硬件·算法
用户47949283569154 小时前
字节Agent开发技术分享—旧会话越聊越笨,新会话又得重讲?我给 Matt Pocock Skill 炼了套《影分身之术》:本体想清楚,分身写代码,完事回来汇报
面试·agent
峥无4 小时前
从0到1手撕红黑树:封装实现 my_map 与 my_set(SGI-STL 源码级深度解析)
开发语言·c++·笔记·算法·stl
波特率1152005 小时前
C++新特性---属性说明符与标准属性
开发语言·c++
不会代码的小猴5 小时前
3. 控件学习1
开发语言·c++·笔记·qt·算法
疯狂打码的少年7 小时前
【数据结构】图的存储结构:邻接矩阵与邻接表
数据结构·笔记
203号居民7 小时前
LeetCode hot 100 —41. 缺失的第一个正数
数据结构·算法·leetcode
蒸蒸yyyyzwd7 小时前
cpp 选手准备秋招学习笔记 day10
笔记·面试·求职招聘