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
相关产品推荐
相关产品推荐

