翻转二叉树

翻转二叉树-力扣

思路

1、递归法-前序遍历
因为涉及左右替换，因此前序遍历(中左右)和后序遍历(左右中)的递归都可

• 递归三部曲

1. 确定递归函数参数和返回值：函数参数root指针，返回值treenode
2. 确定终止条件：root为空终止
3. 确定单层逻辑：前序遍历，先进行交换左右孩子节点，然后反转左子树，反转右子树。

# 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 None
        #中
        root.left,root.right=root.right,root.left
        #左
        self.invertTree(root.left)
        #右
        self.invertTree(root.right)
        return root
        

2、迭代法-前序遍历
用栈，先root入栈，再把右节点入栈，最后把左节点入栈

# 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:
        #前序遍历，迭代法
        if not root:
            return None
        stack=[]
        stack.append(root)
        while stack:
            node=stack.pop()
            #中
            node.left, node.right=node.right, node.left
            #右
            if node.right:
                stack.append(node.right)
            #左
            if node.left:
                stack.append(node.left)
        return root