题目:
给你一棵二叉树的根节点 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 = [] 输出:[]
提示:
- 树中节点数目范围在
[0, 100]
内 -100 <= Node.val <= 100
解题思路:
使用递归的思想,每次翻转当前节点的左右子树
使用迭代的思想,利用队列来存储每一层的节点,交换当前节点的左右节点
递归代码:
class Solution:
def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
if not root:
return None
left = self.invertTree(root.left)
right = self.invertTree(root.right)
root.left = right
root.right = left
return root
迭代代码:
from queue import Queue
# 使用迭代的方法翻转左右子树
class Solution:
def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
if not root:
return None
re_queue = Queue()
re_queue.put(root)
while not re_queue.empty():
current = re_queue.get()
current.left, current.right = current.right, current.left
if current.left:
re_queue.put(current.left)
if current.right:
re_queue.put(current.right)
return root