leetcode226.翻转二叉树 Python

题目:

翻转一棵二叉树。

示例:

输入:

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

输出:

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

思路:

方法1,递归,交换根节点的左右两个分支节点,递归调用翻转函数,实现对整个二叉树的翻转。
方法2,迭代,使用队列翻转二叉树,对当前元素交换左右子树的位置,然后判断左子树是否为空,不为空就放入队列中,判断右子树是否为空,非空就放入队列中,循环让队列元素出队,完成交换,直到队列为空,跳出循环,返回根节点。

# 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 not root:
            return
        root.left ,root.right = root.right,root.left
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root 
        # 迭代
        if not root:
            return 
        queue = [root]
        while queue:
            node = queue.pop(0)
            node.left,node.right = node.right,node.left #交换左右两节点
            if node.left:
                queue.append(node.left)
            if node.right:
                queue.append(node.right)
        return root
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值