• 剑指 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)

  • 相关阅读:
    MUSIC分辨率与克拉美罗下界的关系
    EXCEL 基本函数
    新手如何正确的开始练车
    5.20考试整理
    树上倍增 x
    逆元 x
    BSGS ! x
    【テンプレート】LCA
    [HDOJ5783]Divide the Sequence(贪心)
    [HDOJ5791]Two(DP)
  • 原文地址:https://www.cnblogs.com/yeshengCqupt/p/13448565.html
Copyright © 2020-2023  润新知