LeetCode题解 99.恢复二叉搜索树

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

题目分析：
        对象数据结构为二叉搜索树。二叉搜索树任意节点具有以下特征：节点的左子树只包含小于当前节点的值。节点的右子树只包含大于当前节点值的值。所有左子树和右子树自身必须也是二叉搜索树。因此正确的二叉搜索树中序遍历的结果应该为一递增序列，而当二叉搜索树中仅两个节点的值交换后，其中续遍历的序列中将存在两个点不满足递增关系(称为异常点)，如下图。

        因此可以通过中序遍历得到的序列来判断异常点。为了利用常数空间，这里仅记录中序遍历的前一个点即可，并不需要记录全部序列。可以注意到遍历过程中遇到的第一个异常点将会大于相邻的点，第二个异常点会小于相邻的点。因此对输入树进行中序遍历，遍历过程中比较当前节点值和记录的前一个节点值，第一次出现当前值小于前一点值时，记录前一点为异常点。第二次出现当前值小于前一点值时，记录当前点为异常点。但实际上有可能出现两个异常点在序列中处于相邻位置的情况，即在第一次出现当前值小于前一点值时，当前点和前一点都是异常点的情况，只按照上述逻辑将找不到第二个异常点。因此在第一次出现当前值小于前一点值时，记录前一点异常点，记录当前点为临时异常点，之后，若再次出现异常，更新临时异常点即可。最后交换记录的异常点的值即可。

题目代码：
        最终python代码如下:

class Solution:
    def recoverTree(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        self.x = None
        self.y = None
        self.pre = None

        def travel(root):
            if not root:
                return
            travel(root.left)
            if not self.pre:
                self.pre = root
            else:
                if root.val < self.pre.val:
                    self.y = root
                    if not self.x:
                        self.x = self.pre
                self.pre = root
            travel(root.right)

        travel(root)
        if self.x and self.y:
            self.x.val,self.y.val = self.y.val,self.x.val