【LeetCode练习题】Recover Binary Search Tree

Recover Binary Search Tree

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

 

互换二叉搜索树中两个位置错误的节点。

 

思路：

预设三个指针pre,p1,p2。p1用来指向第一个出错的节点，p2用来指向第二个出错的节点。

出错情况有两种，即p1和p2相邻，p1和p2不相邻。

中序遍历此二叉树，用pre指向当前节点的上一个节点，如果出错的节点相邻，此时p1应该指向pre，p2应该指向当前节点root。

若出错节点不相邻，则用p1记录第一个出错节点pre，继续遍历到前一个节点pre大于当前节点root时，用p2指向第二个出错节点p2.

 

代码如下：

class Solution {
public:
    TreeNode *pre,*p1,*p2;

    void run(TreeNode *root){
        if(!root) 
            return;
        run(root->left);
        if(pre && pre->val > root->val){
            if(p1 == NULL){
                p1 = pre; p2 = root;
            }
            else{
                p2 = root;
            }
        }
        pre = root;
        run(root->right);
    }

    void recoverTree(TreeNode *root) {
        if(!root)
            return ;
        pre = p1 = p2 = NULL;
        run(root);
        swap(p1->val,p2->val);
    }
};