leetcode99. 恢复二叉搜索树

题目

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

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

示例 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 使二叉搜索树有效。

思路

1.因为是二叉搜索树,可以考虑通过中序遍历判断哪些节点有问题
2.注意两种情况

// 1 5 3 2  对换5和2变为 1 2 3 5
// 1 5 3    对换5和3变为 1 3 5

该题需要复习

实现

// 1 5 3 2  对换5和2变为 1 2 3 5
// 1 5 3    对换5和3变为 1 3 5

class Solution {
public:
    TreeNode* a = nullptr;  //第一个逆序的节点
    TreeNode* b = nullptr; //第二个逆序的节点
    TreeNode* c = nullptr; //第三个逆序的节点(可能并不存在)
    TreeNode* pre_node = nullptr;
    void recoverTree(TreeNode* root) {
        func(root);
        if (c) {
            int tmp = a->val;
            a->val = c->val;
            c->val = tmp;
        } else {
            int tmp = a->val;
            a->val = b->val;
            b->val = tmp;
        }
    }

    void func(TreeNode* root) {
        if (root->left) {
            func(root->left);
        }
        if (pre_node && root->val <= pre_node->val) {
            if (!a) {
                a = pre_node;
                b = root;
            } else {
                c = root;
                return;
            }
        }
        pre_node = root;
        if (root->right) {
            func(root->right);
        }
    }
};