leetcode_二叉树 226.翻转二叉树

226.翻转二叉树

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

1. 递归

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def invertTree(self, root):
        """
        :type root: Optional[TreeNode]
        :rtype: Optional[TreeNode]
        """
        if not root:
            return root

        left = self.invertTree(root.left)
        right = self.invertTree(root.right)
        root.left, root.right = right, left
        return root

• 时间复杂度: O(n), n为节点个数
• 空间复杂度: O(h), h是树的高度

2. 深度优先遍历DFS (栈)

def invertTree(root):
    if not root:
        return None

    stack = [root]

    while stack:
        node = stack.pop()
        node.left, node.right = node.right, node.left
        
        # 将左右子节点加入栈
        if node.left:
            stack.append(node.left)
        if node.right:
            stack.append(node.right)
    
    return root

• 时间复杂度: O(n), n为节点个数
• 空间复杂度: O(h), h是树的高度

3. 广度优先遍历BFS (队列)

from collections import deque

def invertTree(root):
    if not root:
        return None

    queue = deque([root])

    while queue:
        node = queue.popleft()
        node.left, node.right = node.right, node.left
        
        # 将左右子节点加入队列
        if node.left:
            queue.append(node.left)
        if node.right:
            queue.append(node.right)
    
    return root

• 时间复杂度: O(n), n为节点个数
• 空间复杂度: O(n),