剑指offer 27 - 二叉树的镜像 - python

58 阅读1分钟

题目描述:

请完成一个函数,输入一个二叉树,该函数输出它的镜像。

例如输入:

     4
   /   \
  2     7
 / \   / \
1   3 6   9
镜像输出:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

示例 1:

输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]

限制:0 <= 节点个数 <= 1000


所谓镜像结构,即当前节点的左右孩子进行互换,直到到达叶节点为止。因为,不断的以左孩子节点和右孩子节点当作root进行子树互换,可采用递归。

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

class Solution:
    def mirrorTree(self, root: TreeNode) -> TreeNode:
        if root is None: 
            return None

        # 交换当前节点的左右孩子节点
        root.left, root.right = root.right, root.left

        self.mirrorTree(root.left)
        self.mirrorTree(root.right)
        
        return root