LeetCode 101. 对称二叉树 | Python

101. 对称二叉树

---

题目

---

给定一个二叉树，检查它是否是镜像对称的。

例如，二叉树 [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

进阶：

你可以运用递归和迭代两种方法解决这个问题吗？

解题思路

---

如何去判断二叉树是镜像对称？如下：

• 根节点为 null，直接返回 True；
• 当根节点不为 null 时，判断左右子树是否对称：
  • 判断 root.left 跟 root.right 节点值是否相同
  • 判断 root.left 的左子树是否与 root.right 的右子树对称
  • 判断 root.left 的右子树是否与 root.right 的左子树对称

按照题目中【进阶】部分所述，本篇幅就从递归和迭代两种思路来解决该问题。

思路：递归

根据上面判断二叉树是否对称的思路，先设定递归终止的条件：

• 如果都为空指针的情况下，返回 True
• 其中一个不为空的情况下，返回 False

递归具体的过程：

• 先判断当前两个指针节点值是否相等
• 判断 t1 的左子树与 t2 的右子树是否对称
• 判断 t1 的右子树与 t1 的左子树是否对称

具体代码实现见【代码实现：递归】。

思路：迭代

还是按照开篇的思路判断二叉树的对称，这次使用迭代的方法实现。这里需要引入一个队列。

具体的做法：

• 初始化的时候，将根节点入队两次。
• 循环的时候，每次出队两次提取两个节点，比较它们的值（因为子树互为镜像，它们的值应该是相等的）
• 再次入队时，将两个节点的左右子节点按照相反的顺序进行入队。（例如： t1 节点的左子节点跟 t2 节点的右子节点连续入队）
• 循环结束的条件，当队列为空的时候，或者连续出队的两个节点不相等时。

具体实现的代码见【代码实现：迭代】

代码实现

---

代码实现：递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:      
        return self.check(root, root)
    
    def check(self, t1, t2):
            if t1 == None and t2 == None:
                return True
            if t1 == None or t2 == None:
                return False
            
            return t1.val == t2.val and self.check(t1.left, t2.right) and self.check(t1.right, t2.left)

代码实现：迭代

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        # 构建，初始化队列
        # 将根节点入队两次
        queue = []
        queue.append(root)
        queue.append(root)

        # 循环开始
        while len(queue) != 0:
            # 先连续出队，提取两个节点进行比较
            t1 = queue.pop()
            t2 = queue.pop()

            if t1 == None and t2 == None:
                continue
            # 当两个节点不相等，直接返回 False
            if t1 == None or t2 == None or t1.val != t2.val:
                return False

            # 重新入队，将两个节点的左右子节点按照相反的顺序入队
            queue.append(t1.left)
            queue.append(t2.right)

            queue.append(t1.right)
            queue.append(t2.left)
        
        return True

实现结果

---

实现结果：递归

实现结果：迭代

总结

---

• 根据题意，按照递归和迭代的思路解决该问题
• 具体的判断是否为镜像对称的思路：
  • 这里判断 root 不为空的情况（根节点为空，直接返回 True）；
  • 判断 root.left 和 root.right 节点值是否相同；
  • 判断 root.left 的左子树和 root.right 的右子树是否对称；
  • 判断 root.left 的右子树和 root.right 的左子树是否对称。
• 递归实现具体的过程：
  • 终止条件：指针都会空时，返回 True；其中一个指针为空时，返回 False；
  • 判断两个指针的节点值是否相同；
  • 判断 t1 的左子树是否与 t2 的右子树对称
  • 判断 t1 的右子树是否与 t2 的左子树对称
• 迭代实现的具体过程：
  • 这里需要引入一个队列，首先初始化队列，先将 root 入队两次；
  • 循环开始，先连续出队两次，提取两个节点，比较它们的值（两个相邻的值应该是相等的，因为左右子树互为镜像）
  • 将两个节点的左右子节点按照相反的顺序进行入队（例如 t1 的左节点与 t2 的右节点连续入队）
  • 循环结束的条件，当队列为空的时候，或者出队的两个节点不相等（也就是不对称）。

---

欢迎关注微信公众号《书所集录》