【leetcode】99 恢复二叉搜索树（二叉树）

题目链接：https://leetcode-cn.com/problems/recover-binary-search-tree/

题目描述

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

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

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

思路

对正确的二叉搜索树进行中序遍历，那么节点系列将会一直升序；如果两个节点的位置错了，一定会出现降序。
（1）如果在中序遍历时节点值出现了两次降序，第一个错误的节点为第一次降序时较大的节点，第二个错误的节点为第二次降序时较小的节点
（1）如果只出现一次降序，则就是这两个节点

采用中序遍历就可以完成题目要求，递归/非递归/Morris遍历都可以。以下采用非递归方式。

复杂度分析
时间复杂度：O（n）
空间复杂度：O（h）

代码

class Solution {
public:
    void recoverTree(TreeNode* root) {
        if(!root) return;
        stack<TreeNode*> st;
        TreeNode *pPrev = nullptr;
        TreeNode *err1 = nullptr, *err2 = nullptr;
        // 非递归中序遍历
        while(!st.empty() || root){
            while(root){
                st.push(root);
                root = root->left;
            }
            root = st.top();
            st.pop();
            if(pPrev && pPrev->val > root->val){
                err1 = err1 ? err1: pPrev;
                err2 = root;
            }
            pPrev = root;
            root = root->right;
        }
        
        swap(err1->val, err2->val);
    }
};