leetcode 99. 恢复二叉搜索树

题目叙述

给你二叉搜索树的根节点 root ，该树中的两个节点被错误地交换。请在不改变其结构的情况下，恢复这棵树。
进阶：使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗？

分析

方法1：$O(n)$的时间复杂度，使用中序遍历

使用中序遍历，先从前往后找第一个逆序的位置，再往后找第二个逆序的位置。然后进行交换。
先存储第一对降序的，后面还有第二对的话，把第二对的第二个数重新赋值修改一下就行了: 即以两个变量来存储一个逆序对，以ans1来存储前者，ans2来存储后者。对于两个逆序对，其实我们最终是要交换第一个逆序对的ans1和第二个逆序对的ans2。
1 5 3 4 2 先存5 3, 后面4 2 又降序了，说明5 2得交换

代码

class Solution {
public:
    //从前往后的中序遍历
    //从后往前的中序遍历
    TreeNode* ans1;
    TreeNode* ans2;
    TreeNode* pre = new TreeNode(INT_MIN);
    void inOrderFront(TreeNode* node) {
        if(!node) return;
        inOrderFront(node->left);
        if(node->val<pre->val){
            if(!ans1) ans1 = pre;//保证ans1只赋值一次
            ans2 = node;
        }
        pre = node;
        inOrderFront(node->right);
    }
    void recoverTree(TreeNode* node) {
        inOrderFront(node);
        //交换两个指针的内容
        int tmp = ans1->val;
        ans1->val = ans2->val;
        ans2->val = tmp;
    }
};