给你二叉搜索树的根节点 root ,该树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。

进阶:使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗?

示例 1:

输入:root = [1,3,null,null,2]
输出:[3,1,null,null,2]
解释:3 不能是 1 左孩子,因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。

示例 2:

输入:root = [3,1,4,null,null,2]
输出:[2,1,4,null,null,3]
解释:2 不能在 3 的右子树中,因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。

提示:

  • 树上节点的数目在范围 [2, 1000] 内
  • -2^31 <= Node.val <= 2^31 – 1

Python 解答:
1.递归

# 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 recoverTree(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        first, second, pre = None, None, TreeNode(-2**31)
        def inOrder(root):
            nonlocal first, second, pre
            if not root:
                return 
            inOrder(root.left)
            if root.val < pre.val:
                if not first:
                    first = pre
                second = root
            pre = root
            inOrder(root.right)
        inOrder(root)
        first.val, second.val = second.val, first.val
最后修改日期: 2021年8月3日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。