【Leetcode】Python实现—226.翻转二叉树


一、题目描述

翻转一棵二叉树。

示例:

输入:
在这里插入图片描述

输出:
在这里插入图片描述

二、递归方法

思路与算法

我们在做二叉树题目时候,第一想到的应该是用 递归 来解决。
仔细看下题目的 输入输出,输出的左右子树的位置跟输入正好是相反的,于是我们可以递归的交换左右子树来完成这道题。
在这里插入图片描述
其实就是交换一下左右节点,然后再递归的交换左节点,右节点。根据动画图我们可以总结出递归的两个条件如下:

终止条件:当前节点为 null 时返回
递归条件:交换当前节点的左右节点,再递归的交换当前节点的左节点,递归的交换当前节点的右节点

代码实现

class Solution(object):
	def invertTree(self, root):
		"""
		:type root: TreeNode
		:rtype: TreeNode
		"""
		# 递归函数的终止条件,节点为空时返回
		if not root:
			return None
		# 将当前节点的左右子树交换
		root.left,root.right = root.right,root.left
		# 递归交换当前节点的 左子树和右子树
		self.invertTree(root.left)
		self.invertTree(root.right)
		# 函数返回时就表示当前这个节点,以及它的左右子树
		# 都已经交换完了		
		return root

复杂度分析

**时间复杂度:**每个元素都必须访问一次,所以是 O ( n ) O(n) O(n)
**空间复杂度:**最坏的情况下,需要存放 O ( h ) O(h) O(h) 个函数调用( h h h是树的高度),所以是 O ( h ) O(h) O(h)

三、迭代方法

思路与算法

递归实现也就是深度优先遍历的方式,那么对应的就是广度优先遍历。
广度优先遍历需要额外的数据结构–队列,来存放临时遍历到的元素。
深度优先遍历的特点是一竿子插到底,不行了再退回来继续;而广度优先遍历的特点是层层扫荡。
所以,我们需要先将根节点放入到队列中,然后不断的迭代队列中的元素。
对当前元素调换其左右子树的位置,然后:

判断其左子树是否为空,不为空就放入队列中
判断其右子树是否为空,不为空就放入队列中

动态图如下:
在这里插入图片描述

代码实现

class Solution(object):
	def invertTree(self, root):
		"""
		:type root: TreeNode
		:rtype: TreeNode
		"""
		if not root:
			return None
		# 将二叉树中的节点逐层放入队列中,再迭代处理队列中的元素
		queue = [root]
		while queue:
			# 每次都从队列中拿一个节点,并交换这个节点的左右子树
			tmp = queue.pop(0)
			tmp.left,tmp.right = tmp.right,tmp.left
			# 如果当前节点的左子树不为空,则放入队列等待后续处理
			if tmp.left:
				queue.append(tmp.left)
			# 如果当前节点的右子树不为空,则放入队列等待后续处理	
			if tmp.right:
				queue.append(tmp.right)
		# 返回处理完的根节点
		return root

处理的是tmp, 最后返回的是root, 修改tmp 会影响root的内存
tmp和root都是引用同一个对象的,类似于C++的指针,多个指针可以指向同一块内存。
至于为什么不直接操作root,是为了防止丢失root指针。

复杂度分析

**时间复杂度:**同样每个节点都需要入队列/出队列一次,所以是 O ( n ) O(n) O(n)
**空间复杂度:**最坏的情况下会包含所有的叶子节点,完全二叉树叶子节点是 n / 2 n/2 n/2个,所以时间复杂度是 O ( n ) O(n) O(n)

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值