Recover Binary Search Tree-恢复二叉查找树

题目描述:

由于某种原因一个二叉排序树的两个节点的元素被交换,在不改变树的结构的情况下恢复这颗二叉排序树

题目来源:

http://oj.leetcode.com/problems/recover-binary-search-tree/

题目分析:

中序遍历二叉排序树会得到递增序列,如果两个元素交换,递增序列的顺序必被破坏,例如:1, 2, 3, 4, 5, 6, 7。交换后可能会有

(1)1, 2, 6, 4, 5, 3, 7 (2)1, 2, 3, 4, 65,7 等

在中序遍历时,出现当前元素比中序遍历中前一个元素小时,找到出错元素,维护一个刚刚遍历的节点的指针pre,当遍历结束当前节点时,更新pre指针

时间复杂度:O(1)

示例代码:
TreeNode *pre, *first, *second;

void recoverTree(TreeNode *root) {
    pre = first = second = NULL;
    inOrder(root);
    int tmp = second->val;
    second->val = first->val;
    first->val = tmp;
}

void inOrder(TreeNode*& root) {
    if(root == NULL)
        return ;

    inOrder(root->left);
    if(pre != NULL && pre->val > root->val) {
        first == NULL ? first = pre, second = root : second = root;
    }
    pre = root;
    inOrder(root->right);

    return ;
}

 

posted @ 2013-09-25 20:13  代金桥  阅读(591)  评论(0编辑  收藏  举报