Leetcode 99. Recover Binary Search Tree （python+cpp）

最新推荐文章于 2022-07-27 13:46:02 发布

努利！奋斗！

最新推荐文章于 2022-07-27 13:46:02 发布

阅读量203

点赞数

分类专栏： Leetcode 树文章标签： c++ leetcode python BST

本文链接：https://blog.csdn.net/qq_37821701/article/details/107134368

版权

本文介绍LeetCode 99题的解决方案，通过中序遍历发现二叉搜索树中被交换的两个节点，然后修复树的结构。提供Python和C++两种实现方式。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

解析：

这道题目的解法有很多，但是我在这边只介绍一种比较基本的也最容易理解的方法。解这道题的关键在于，二叉查找树的中序遍历是一个单调递增的序列。所以把错误的二叉查找树复原之后，第一个错误位置的元素会比他后一个元素大，而第二个错误的元素会比他前面的元素小，根据这样的规律就可以找出两个元素，所以具体的步骤如下：

将原树进行中序遍历
找出被交换的两个元素
遍历原树，修改相应两个位置的值

python代码如下：

    def recoverTree(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        def inorder(root):
            return inorder(root.left)+[root.val]+inorder(root.right) if root else []
        
        def find_swap(nums):
            x = y = -1
            for i in range(len(nums)-1):
                if nums[i+1]<nums[i]:
                    y = nums[i+1