leetcode练习题 recover-binary-search-tree

解题思路

将有两个结点出错的二叉搜索树恢复为正确的二叉搜索树。用pre保存先序遍历的上一个结点，用a和b保存要交换的两个结点，若pre不为空且pre的值大于等于当前根节点的值，则修改b的值，当a为空修改a的值，并将pre修改为当前根结点，之后先序遍历根结点的右子树，最后若a、b不为空，则交换a和b的值。

代码

class Solution {
public:
    void preorder(TreeNode *root,TreeNode *&pre,TreeNode *&a,TreeNode *&b){
        if(root != NULL){
            preorder(root->left,pre,a,b);
            if(pre && pre->val >= root->val){
                if(!a)
                    a = pre;
                b = root;
            }
            pre = root;
            preorder(root->right,pre,a,b);
        }
    }
    void recoverTree(TreeNode *root) {
        TreeNode *pre = NULL,*a = NULL,*b = NULL;
        preorder(root,pre,a,b);
        if(a && b){
            int tmp = a->val;
            a->val = b->val;
            b->val = tmp;
        }
    }
};