99. 存在两个错误交换结点的二叉搜索树

class Solution(object):

    # 中序遍历会发现一个或两个降序点
    # 示例: [2,4,3,5](一个降序点); [5,3,4,2](两个降序点)
    # 把降序点找到然后把他们的值交换
    # 用全局变量保存前一个结点,第一个要交换的点和第二个要交换的点,以及记录第几次发现降序点
    # 如果是一个降序点的话,要交换的值是发现降序点位置的前一个和后一个
    # 如果是两个降序点的话,交换的值是第一次降序点的前一个和第二次降序点的后一个

    def __init__(self):
        self.firstNode = None
        self.secondNode = None
        self.pre = None
        self.flag = True

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

        self.inOrder(root)
        self.firstNode.val, self.secondNode.val = self.secondNode.val, self.firstNode.val

    def inOrder(self, root):
        if not root:
            return

        # 中序遍历
        self.inOrder(root.left)
        if self.pre and self.pre.val > root.val:
            if self.flag:
                self.firstNode = self.pre
                self.secondNode = root
                self.flag = False
            else:
                self.secondNode = root
        # 当前步执行完之后设置pre = root
        self.pre = root
        self.inOrder(root.right)