leetCode第226题 翻转二叉树
链接:https://leetcode-cn.com/problems/invert-binary-tree
给你一棵二叉树的根节点 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
翻转二叉树
一开始我还以为时像对称二叉树一样递归地把左节点地左孩子和右节点地右孩子交换并把左结点的右孩子和右节点的左孩子交换(其实也没毛病),但实际上自底向上地把树做一个交换就行了。
代码非常简单,同样用递归实现
当root为None,返回None
否则调用自身,先把左子树翻转
调用自身,把右子树翻转
然后把左右子树翻转并返回根节点
## python3
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: TreeNode) -> TreeNode:
if root == None:
return None
self.invertTree(root.left)
self.invertTree(root.right)
temp = root.left
root.left = root.right
root.right = temp
return root