剑指 Offer 28. 对称的二叉树 - leetcode 剑指offer系列

题目难度: 简单

原题链接

今天继续更新剑指 offer 系列, 这道题同样可以用递归或迭代来做, 这里提供两种递归和一种迭代的方法帮助大家拓展思路

老样子晚上 6 点 45 分准时更新公众号 每日精选算法题, 大家记得关注哦~ 另外在公众号里回复 offer 就能看到剑指 offer 系列当前连载的所有文章了

题目描述

请实现一个函数，用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样，那么它是对称的。

例如，二叉树  [1,2,2,3,4,4,3] 是对称的。

    1
   / \
  2   2
 / \ / \
3  4 4  3

但是下面这个  [1,2,2,null,3,null,3] 则不是镜像对称的:

    1
   / \
  2   2
   \   \
   3    3

• 0 <= 节点个数 <= 1000

题目样例

示例 1

输入

root = [1,2,2,3,4,4,3]

输出

true

示例 2

输入

root = [1,2,2,null,3,null,3]

输出

false

题目思考

1. 生成镜像比较可以吗, 直接比较呢?
2. 如果限制只能用递归或者迭代, 如何解决?

解决方案

方案 1

思路

• 根据题目描述, 最自然的思路就是生成原树的镜像后, 再逐个节点比较是否完全相等即可
• 对于生成镜像部分不清楚的同学可以回顾下上篇文章 剑指 Offer 27. 二叉树的镜像 - leetcode 剑指offer系列
• 需要注意这里生成的镜像不能直接在原树基础上改动了, 需要生成新的节点, 也即新建一个节点使用原来的节点值, 不然下面比较的时候就是和自身比较, 肯定总是相等了
• 逐个节点比较也很简单, 如果其中某个节点为空, 则必须另一个也是空; 而如果两个节点都非空的话, 则要比较它们的值, 并传入左右子节点递归比较子树的相等性

复杂度

• 时间复杂度 O(N)
  • 所有节点只需要遍历一遍
• 空间复杂度 O(N)
  • 递归栈的消耗, 以及镜像树的空间

代码

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        # 方法1: 先拿到镜像, 然后逐节点比较
        def getMirror(root):
            if not root:
                return root
            # 注意这里要新建node, 不能原地修改!!
            newNode = TreeNode(root.val)
            newNode.left, newNode.right = getMirror(root.right), getMirror(
                root.left)
            return newNode

        mirror = getMirror(root)

        def ismatch(n1, n2):
            if not n1 or not n2:
                return not n1 and not n2
            # 这里需要比较n1和n2树精确匹配
            return n1.val == n2.val and ismatch(n1.left, n2.left) and ismatch(
                n1.right, n2.right)

        # 当原树和镜像树完全匹配的时候才对称
        return ismatch(root, mirror)

方案 2

思路

• 回顾方案 1, 我们需要先生成镜像再比较, 有没有直接一遍比较就能得出结果的方法呢?
• 答案也是有的, 注意题目描述, 对称时左子树的最左节点等于右子树的最右节点, 就像中间有个对称轴一样
• 所以我们可以传入当前左右子树上处于对称位置的两个节点, 比较它们是否相等, 相等的话再递归调用比较方法, 比较(左节点的右子节点, 右节点的左子节点)以及(左节点的左子节点, 右节点的右子节点), 也即子树上的两对对称节点, 如果都满足对称性的话则说明整个树对称

复杂度

• 时间复杂度 O(N)
  • 所有节点只需要遍历一遍
• 空间复杂度 O(N)
  • 递归栈的消耗

代码

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        # 方法2: 递归直接比较
        if not root:
            return True

        def symm(left, right):
            if not left or not right:
                # left和right某一个是空的时候, 需要两个都是空才算对称
                return not left and not right
            # 对称需要满足当前两节点值相等, 且左边的左节点=右边的右节点 & 左边的右节点=右边的左节点
            return left.val == right.val and symm(
                left.left, right.right) and symm(left.right, right.left)

        # 此时root一定非空, 直接传入左右子节点判断对称性即可
        return symm(root.left, root.right)

方案 3

思路

• 接下来我们考虑迭代做法
• 根据方案 2, 我们需要比较左右子树处于对称位置的节点
• 所以我们完全可以将这些对称节点放在两个队列中, 分别代表左右子树中的节点
• 然后仍然按照方案 2 的顺序将子节点依次加入队列, 保证比较的是处于对称位置的节点即可
• 注意空节点也需要被加入队列, 用作占位符, 不然对于用例 2 就会错误地判断成对称的了
• 下面代码中有详细解释, 方便大家理解

复杂度

• 时间复杂度 O(N)
  • 所有节点只需要遍历一遍
• 空间复杂度 O(N)
  • 两个队列各自需要存 N/2 个节点

代码

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        # 方法3: 迭代, 使用两个队列存左子树和右子树的节点
        if not root:
            return True
        leftq, rightq = [root.left], [root.right]
        # 这里可以优化为只使用一个下标, 因为下标总是相等的. 这里为了更容易理解, 采用两个下标
        i, j = 0, 0
        while i < len(leftq) or j < len(rightq):
            if i == len(leftq) or j == len(rightq):
                # 其中一方队列遍历完了, 肯定不对称
                return False
            left, right = leftq[i], rightq[j]
            i += 1
            j += 1
            if not left or not right:
                # 某一边节点为空的情况
                if not left and not right:
                    # 必须两边都是空才可以继续循环
                    continue
                else:
                    # 否则肯定不对称
                    return False
            else:
                if left.val != right.val:
                    # 值不相等, 不对称
                    return False
                # 注意这里顺序, 保证左边的right和右边的left比较, 左边的left和右边的right比较 (同方法2思路)
                leftq.append(left.right)
                leftq.append(left.left)
                rightq.append(right.left)
                rightq.append(right.right)
        return True

---

大家可以在下面这些地方找到我~😊

我的知乎专栏

我的 CSDN

我的 Leetcode

我的牛客网博客

我的公众号: 每日精选算法题, 欢迎大家扫码关注~😊