You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Python交换操作实现翻转二叉树的问题求助

翻转二叉树递归实现的两种写法差异

昨晚在LeetCode处理二叉树相关题目时,遇到翻转二叉树问题,尝试用递归结合Python交换操作实现,但两种写法结果不同:

已通过的写法

# 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: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None

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

        return root

未通过的写法

# 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: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None

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

        return root

核心差异在于两种交换方式的执行逻辑:

交换方式1(正确)

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

Python的元组同时赋值会先计算右侧所有表达式的结果,将self.invertTree(root.right)和self.invertTree(root.left)的结果存入临时元组,再一次性赋值给root.left和root.right。这两个递归调用都是基于原始root节点的左右子树执行的,彼此互不影响,能正确完成左右子树的翻转交换。

交换方式2(错误)

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

这是分步赋值:第一步先把root.left替换为翻转后的右子树,此时root.left已经不是原始的左子树了。第二步执行self.invertTree(root.left)时,实际是对第一步修改后的新root.left(即原右子树的翻转结果)再次翻转,最终root.right得到的是原始的root.right,相当于没完成交换,甚至破坏了树的结构,导致结果错误。

内容的提问来源于stack exchange,提问作者PhilipSanM

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.20 12:15:02