leetcode 99. 恢复二叉搜索树（C++）

题目

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

示例 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) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗？

来源：99. 恢复二叉搜索树 - 力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
算法来自官方题解——>恢复二叉搜索树 - 恢复二叉搜索树 - 力扣（LeetCode）

中序遍历法

• 若要找到交换的节点，就记录中序遍历中的最后一个节点 pred（即当前节点的前置节点），并与当前节点的值进行比较。如果当前节点的值小于前置节点 pred 的值，说明该节点是交换节点之一。
  交换的节点只有两个，因此在确定了第二个交换节点以后，可以终止遍历。

代码如下：

class Solution {
public:
    void recoverTree(TreeNode* root) {
        if(!root)   return ;
        vector<TreeNode*> stack;
        TreeNode* x = NULL;
        TreeNode* y = NULL;
        TreeNode* pred = NULL;
        while(!stack.empty() || root){
            while(root){
                stack.push_back(root);
                root = root->left;
            }
            root = stack.back();
            stack.pop_back();
            if(pred && root->val < pred->val){
                y = root;
                if(!x)  x = pred;
                else    break;
            }
            pred = root;
            root = root->right;
        }
        int temp = x->val;
        x->val = y->val;
        y->val = temp;
    }
};

结果：