LeetCode 99. 恢复二叉搜索树

题目链接 LeetCode 99. 恢复二叉搜索树

题目描述

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

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

解题思路

这个题我们可以用到二叉排序树的特殊性质，因为二叉排序树的中序遍历是有序性的，而且本题只有两个位置的数字被交换了，所以我们在中序遍历的时候存一个当前节点的前驱节点，如果不满足前驱节点的数值小于当前节点，就说明被调换了。这里不用担心为什么只比较这两个节点，因为如果是当前节点的前几个前驱中的某一个比当前节点大，说明就不止调换了两个元素。

程序代码

c++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *x = NULL, *y = NULL, *pred = NULL;
    void swap(TreeNode* a, TreeNode* b) {
        int tmp = a->val;
        a->val = b->val;
        b->val = tmp;
    }
    void findTwoSwapped(TreeNode* root) {
        if (root ==　NULL) return;
        findTwoSwapped(root->left);
        if (pred != NULL && root->val < pred->val) {
            y = root;//记录当前节点
            if (x == NULL) x = pred;//记录前驱节点
            else return;
            //这一句else可以不用，因为只有一处不满足，只要进入当前这个大的if里面，就不会在进入第二次了
        }
        pred = root;//记录前驱节点
        findTwoSwapped(root->right);
    }
    void recoverTree(TreeNode* root) {
        findTwoSwapped(root);
        swap(x,y);
    }
};