剑指 Offer 27. 二叉树的镜像(递归/辅助栈)

• 题目描述

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

例如输入：

     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

• 解法一：递归解法

思路很简单，直接递归调用将左右子树进行交换就行，这里如果不是python直接交换赋值的话，需要用一个tmp将左子树保存，不然随着递归左子树的值会改变。

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
class Solution:
    def mirrorTree(self, root: TreeNode) -> TreeNode:
        if not root: return
        tmp = root.left
        root.left = self.mirrorTree(root.right)
        root.right = self.mirrorTree(tmp)
        return root

时间复杂度O(N)，空间复杂度O(N)

• 解法二:用辅助栈

利用栈先进后出的特性，将node的左右子树分别压入栈，先压入左子树入栈，然后再压入右子树入栈，然后每次以出栈的栈顶的左右子树作为新的node入栈，这样则可以一直将右子树入栈，然后每次入栈就进行root的左右子树交换。

这里栈是用来辅助从根节点往后遍历，然后指定交换的左右子树的。我觉得还是比较难想到！

    def mirrorTree2(self, root: TreeNode) -> TreeNode:
        if not root:
            return
        stack = [root]
        while stack:
            node = stack.pop()
            if node.left:
                stack.append(node.left)
            if node.right:
                stack.append(node.right)
            node.left, node.right = node.right, node.left
        return root

时间复杂度O(N)，空间复杂度O(N)