Leetcode 226. 翻转二叉树

题目描述:
给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。

示例 1:
在这里插入图片描述
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]

示例 2:
在这里插入图片描述
输入:root = [2,1,3]
输出:[2,3,1]
示例 3:
输入:root = []
输出:[]

思路:
解法1:递归

1、终止条件: 当节点 rootrootroot 为空时(即越过叶节点),则返回 nullnullnull 。
2、递推工作:
3、初始化节点 tmptmptmp ,用于暂存 rootrootroot 的左子节点。
开启递归 右子节点 invertTree(root.right)invertTree(root.right)invertTree(root.right) ,并将返回值作为 rootrootroot 的 左子节点 。
开启递归 左子节点 invertTree(tmp)invertTree(tmp)invertTree(tmp) ,并将返回值作为 rootrootroot 的 右子节点 。
4、返回值: 返回当前节点 rootrootroot 。
注意:为何需要暂存 rootrootroot 的左子节点?
在递归右子节点 “root.left=invertTree(root.right);root.left = invertTree(root.right);
root.left=invertTree(root.right);” 执行完毕后, root.leftroot.leftroot.left 的值已经发生改变,此时递归左子节点 invertTree(root.left)invertTree(root.left)invertTree(root.left) 则会出问题。

python:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root:
            return
        tmp=root.left
        root.left=self.invertTree(root.right)
        root.right=self.invertTree(tmp)
        return root

复杂度分析:
时间复杂度 O(N): 其中 N 为二叉树的节点数量,建立二叉树镜像需要遍历树的所有节点,占用 O(N)时间。
空间复杂度 O(N): 最差情况下(当二叉树退化为链表),递归时系统需使用 O(N) 大小的栈空间。

解法2:栈
利用栈遍历树的所有节点 node,并交换每个 node的左 / 右子节点。

步骤:
1、 当 root为空时,直接返回 null。
2、初始化:用栈,并加入根节点 root 。
3、循环交换: 当栈 stack 为空时跳出。
出栈: 记为 node。
添加子节点: 将 node左和右子节点入栈。
交换: 交换 node 的左 / 右子节点。
4、返回值: 返回根节点 root。

python:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def invertTree(self, root: Optional[TreeNode]) -> Optional[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

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值