leetcode：226. 翻转二叉树【dfs翻转】

分析

dfs返回这个子树反转后的结果
如果是叶子节点就返回自己
如果有左孩子，右孩子就变成反转后的左孩子
如果有右孩子，左孩子就变成反转后的右孩子
然后要用temp中间记录一下，不能边改边dfs
temp初值为None即可

ac code

# Definition for a binary tree node.
# 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 not root:
            return root
        def dfs(node):
            if node.left is None and node.right is None:
                return node
            
            temp1 = temp2 = None
            
            if node.left:
                temp1 = dfs(node.left)
            if node.right:
                temp2 = dfs(node.right)
            

            node.right = temp1
            node.left = temp2
            
            return node
        
        dfs(root)
        return root

总结

反转二叉树用了dfs也是一道很经典的题