leetcode 99 恢复二叉树

99. 恢复二叉搜索树
    二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下，恢复这棵树。

示例 1:

输入: [1,3,null,null,2]

1
/
3

2

输出: [3,1,null,null,2]

3
/
1

2
示例 2:

输入: [3,1,4,null,null,2]

3
/
1 4
/
2

输出: [2,1,4,null,null,3]

2
/
1 4
/
3
进阶:

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

/**
 * 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 {
    vector<pair<int,TreeNode*> >v;
public:
    void dfs(TreeNode *root){
        if(root->left){
            dfs(root->left);
        }
        pair temp(root->val,root);
        v.push_back(temp);
        if(root->right){
            dfs(root->right);
        }
        return ;
    }
    void recoverTree(TreeNode* root) {
        if(root){
            dfs(root);
            int pos1=-1,pos2=-1;
            for(int i=0;i<v.size()-1;i++){
                if(v[i].first>v[i+1].first){
                    if(pos1==-1){
                        pos1=i;
                        pos2=i+1;
                    }
                    else {
                        pos2=i+1;
                        break;
                    }
                }
            }
            swap(v[pos1].second->val,v[pos2].second->val);
        }
    }
};