LeetCode226. Invert Binary Tree

最新推荐文章于 2024-10-29 00:15:00 发布

原创最新推荐文章于 2024-10-29 00:15:00 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文详细介绍了颠倒二叉树的算法实现，包括递归方法、BFS和DFS三种方式，通过实例展示了如何将一棵二叉树的左右子树进行交换，达到颠倒整棵树的效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Invert a binary tree.

Example:

Input:

     4
   /   \
  2     7
 / \   / \
1   3 6   9

Output:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

思路:颠倒二叉树,用递归方法很简单.

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None
        
        root.left, root.right = root.right, root.left
        
        self.invertTree(root.left)
        self.invertTree(root.right)
        
        return root

当然也可以用BFS和DFS去做

# BFS
def invertTree(self, root):
    queue = collections.deque([(root)])
    while queue:
        node = queue.popleft()
        if node:
            node.left, node.right = node.right, node.left
            queue.append(node.left)
            queue.append(node.right)
    return root

# DFS
def invertTree(self, root):
    stack = [root]
    while stack:
        node = stack.pop()
        if node:
            node.left, node.right = node.right, node.left
            stack.extend([node.right, node.left])
    return root