题目描述:
由于某种原因一个二叉排序树的两个节点的元素被交换,在不改变树的结构的情况下恢复这颗二叉排序树
题目来源:
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, 6, 5,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 ; }