翻转二叉树-python

leetCode第226题 翻转二叉树
链接:https://leetcode-cn.com/problems/invert-binary-tree

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

示例 1:

翻转二叉树-python_第1张图片
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]

示例 2:

翻转二叉树-python_第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

你可能感兴趣的:(算法刷题,python,leetcode,算法)