99. 恢复二叉搜索树

2021/03/02 leetcode 共 818 字,约 3 分钟

title

problem link

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

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

示例 1:

img

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

示例 2:

img

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

提示:

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

solution

class Solution:
    left, right, pre = None, None, None

    def recoverTree(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        self.inorder(root)
        if self.left and self.right:
            self.left.val, self.right.val = self.right.val, self.left.val
        return

    def inorder(self, root: TreeNode):
        if not root:
            return
        if root.left:
            self.inorder(root.left)
        # visit
        if self.pre and root.val < self.pre.val:
            self.right = root
            if self.left is None:
                self.left = self.pre
        self.pre = root
        if root.right:
            self.inorder(root.right)
        return

Search

    Table of Contents