(补)算法训练Day15 | LeetCode226. 翻转二叉树(10种遍历方式的应用);LeetCode101. 对称二叉树(考查遍历两个二叉树)

目录

LeetCode226. 翻转二叉树

1. 思路

2. 代码实现

2.1 递归法实现(前中后序)

2.2 迭代法(前中后序)

 2.3 统一迭代法

2.4 层序遍历实现

3. 复杂度分析

4. 思考

LeetCode101. 对称二叉树

方法一:递归解法

1. 思路

2. 代码实现

3. 复杂度分析

4. 思考

方法二:迭代解法

1. 思路

2. 代码实现

实现方式一:使用队列迭代

实现方式二:使用栈迭代

3. 复杂度分析

4. 思考


LeetCode226. 翻转二叉树

链接:  226. 翻转二叉树 - 力扣(LeetCode)

1. 思路

我们之前介绍的都是各种方式遍历二叉树,这次要翻转了,感觉还是有点懵逼。

这得怎么翻转呢?如果要从整个树来看,翻转还真的挺复杂,整个树以中间分割线进行翻转,如图:

可以发现想要翻转它,其实就把每一个节点的左右孩子交换一下就可以了。

关键在于遍历顺序,前中后序应该选哪一种遍历顺序? (一些同学这道题都过了,但是不知道自己用的是什么顺序)

遍历的过程中只要把每一个节点的左右孩子翻转一下,就可以达到整体翻转的效果。这道题目使用前序遍历和后序遍历都可以,唯独中序遍历不方便,因为中序遍历(递归法)会把某些节点的左右孩子翻转了两次!建议拿纸画一画,就理解了。那么层序遍历可以不可以呢?依然可以的!只要把每一个节点的左右孩子翻转一下的遍历方式都是可以的!

中序遍历为啥不行?

以上面的二叉树为例,中序为左中右,先处理4的左孩子,把1和3交换顺序之后,再处理中节点,把2及其子树和7及其子树交换位置,现在该处理右节点了,但是发现右节点是之前处理过了的节点2及其子树,所以节点2及其子树会交换两遍,节点7及其子树没有处理。

2. 代码实现

用遍历二叉树的各种方法都可以实现,一共有10种。

包括递归法(前中后序)迭代法(前中后序)统一迭代法(前中后序)层序遍历

这里以前序递归为例实现:

# 前序递归实现 翻转二叉树 中左右
# time:O(N);space:O(N)
class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return # base case
        root.left, root.right = root.right,root.left #中
        self.invertTree(root.left) # 左
        self.invertTree(root.right) # 右
        return root

2.1 递归法实现(前中后序)

# 翻转二叉树 递归法
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right

# 递归 前序遍历

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return None 
        root.left,root.right = root.right,root.left
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root 

# 递归 后序遍历

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return None
        self.invertTree(root.left)
        self.invertTree(root.right)
        root.left,root.right = root.right,root.left
        return root

# 递归 中序遍历

# time:O(N)  space:O(logN)
class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return None
        self.invertTree(root.left)
        root.left,root.right = root.right,root.left
        # 这里还是要翻转左孩子,因为上面左右互换了
        self.invertTree(root.left)
        return root

2.2 迭代法(前中后序)

#  迭代法 前序遍历

# time:O(N);space:O(N)
class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return None
        stack = [root]
        while stack:
            node = stack.pop()
            node.left,node.right = node.right,node.left
            if node.right: stack.append(node.right)
            if node.left: stack.append(node.left)
        return root

# 迭代法 中序遍历

# time O(N) space:O(N)
class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root == None: return None
        stack = []
        cur = root
        while cur or stack:
            if cur:
                stack.append(cur)
                cur = cur.left
            else:
                cur = stack.pop()
                cur.left,cur.right = cur.right,cur.lef
  • 5
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值