99. 恢复二叉搜索树

99. 恢复二叉搜索树

题目介绍

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

示例 1：

输入：root = [1,3,null,null,2]
输出：[3,1,null,null,2]
解释：3 不能是 1 的左孩子，因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
示例 2：

输入：root = [3,1,4,null,null,2]
输出：[2,1,4,null,null,3]
解释：2 不能在 3 的右子树中，因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。

提示：

树上节点的数目在范围 [2, 1000] 内
-231 <= Node.val <= 231 - 1

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

中序遍历

对于[1,2,3,4,5] 的树，如果两个点被交换变为 [1,4,3,2,5]，我们发现会有两个部分在结点与上一个结点的大小关系被改变，因此我们只需要找到这两个结点即可。第一个部分是需要记录上一个结点，第二个部分需要记录当前结点。

也有可能只有一个部分的顺序被改变，因此遇到第一个结点时更新 first，而遇到的每一个结点都去更新 second。

/**
 * 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:
    TreeNode* first = NULL; // 记录第一个出错的结点
    TreeNode* second = NULL; // 记录第二个出错的结点
    TreeNode* last_node = NULL; // 记录中序遍历的上一个结点
    long last_val = LONG_MIN;

    void dfs(TreeNode* root) {
        if(root==NULL) return;
        dfs(root->left);
        if(root->val < last_val){
            if(first==NULL) first = last_node;
            second = root;
        }
        last_val = root->val;
        last_node = root;
        dfs(root->right);
    }
    void recoverTree(TreeNode* root) {
        dfs(root);
        int s = first->val;
        first->val = second->val;
        second->val = s;
    }
};

时间复杂度为$O(n)$，空间复杂度为$O(k)$，其中 $k$ 为二叉树的深度。