0
点赞
收藏
分享

微信扫一扫

leetcode-99. 恢复二叉搜索树

陆佃 2022-02-14 阅读 52
leetcode
99. 恢复二叉搜索树
给你二叉搜索树的根节点 root ,该树中的 恰好 两个节点的值被错误地交换。请在不改变其结构的情况下,恢复这棵树 。

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二叉搜索树的中序遍历是递增的 错误交换说明在中序遍历中存在非递增的节点
因为刚好是两个节点被错误地交换 所以有两个非递增地节点 找到这两个节点 交换值即可

class Solution:
    def recoverTree(self, root: Optional[TreeNode]) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        last_node = None
        maxer_node = None
        miner_node = None
        def mid_deep(root):
            nonlocal last_node
            nonlocal maxer_node
            nonlocal miner_node
            if root is None:
                return
            mid_deep(root.left)
            if last_node is not None and last_node.val > root.val:
                if maxer_node is None:
                    maxer_node = last_node
                miner_node = root
            last_node = root
            mid_deep(root.right)
        mid_deep(root)
        tmp = maxer_node.val 
        maxer_node.val = miner_node.val
        miner_node.val = tmp

举报

相关推荐

0 条评论