Leetcode 99 恢复二叉搜索树

题目描述：

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

进阶：使用 O(n) 空间复杂度的解法很容易实现。你能想出一个只使用常数空间的解决方案吗？

示例：

 

 输入：

[1,3,null,null,2]

输出：

[3,1,null,null,2] 

思路：

因为二叉搜索树的中序遍历是一个有序数组。

根据题目描述，上述示例的中序遍历为 [3，2，1] 。

所以第一个节点的值和第三个节点的值应该互换。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> tmp;

    void inorder(TreeNode* node){
        if (!node) return;
        inorder(node->left);
        tmp.push_back(node);
        inorder(node->right);
    }

    void recoverTree(TreeNode* root) {
        inorder(root);

        TreeNode *fir=nullptr;
        TreeNode *sec=nullptr;

        for (int i = 0; i < tmp.size() - 1; ++i) {
            if (tmp[i]->val > tmp[i + 1]->val) {
                sec = tmp[i + 1];
                if (fir==NULL) {
                    fir = tmp[i];
                }
            }
        }
        if(fir!=nullptr&&sec!=nullptr)
        {
            int ndtmp = fir->val;
            fir->val = sec->val;
            sec->val = ndtmp;
        }
    }
};