Two elements of a binary search tree (BST) are swapped by mistake.
Recover the tree without changing its structure.
Note:A solution using O( n ) space is pretty straight forward. Could you devise a constant space solution?
confused what "{1,#,2,3}"
means? > read more on how binary tree is serialized on OJ.
Subscribe to see which companies asked this question
一、题目描述
一棵二叉树原本是二叉搜索树,但是其中有两个节点调换了位置,要求恢复原来的二叉搜索树
二、解题思路
中序遍历二叉树,出现的节点的值会升序排序,如果有两个节点位置错了,那肯定会出现降序。
设置一个pre节点指针,记录当前节点中序遍历时的前节点,如果当前节点小于pre节点的值,说明需要调整次序。
如果在中序遍历时节点值出现了两次降序,第一个错误的节点为第一次降序时较大的节点,第二个错误节点为第二次降序时较小的节点。
比如,原来的搜索二叉树在中序遍历的节点值依次为{1,2,3,4,5},如果因为两个节点位置错了而出现{1,5,3,4,2},
第一次降序为5->3,所以第一个错误节点为5,第二次降序为4->2,所以第二个错误节点为2,将5和2换过来就可以恢复。
//中序遍历二叉树,出现的节点的值会升序排序,如果有两个节点位置错了,那肯定会出现降序。
//设置一个pre节点指针,记录当前节点中序遍历时的前节点,如果当前节点小于pre节点的值,说明需要调整次序。
//如果在中序遍历时节点值出现了两次降序,第一个错误的节点为第一次降序时较大的节点,第二个错误节点为第二次降序时较小的节点。
//比如,原来的搜索二叉树在中序遍历的节点值依次为{1,2,3,4,5},如果因为两个节点位置错了而出现{1,5,3,4,2},
//第一次降序为5->3,所以第一个错误节点为5,第二次降序为4->2,所以第二个错误节点为2,将5和2换过来就可以恢复。
class Solution {
public:
TreeNode* mistake1;
TreeNode* mistake2;
TreeNode* pre=NULL;
void recoverTree(TreeNode* root) {
recursive_traversal(root);
if (mistake1 != NULL && mistake2 != NULL)
{
swap(mistake1->val, mistake2->val);
}
}
//递归中序遍历二叉树
void recursive_traversal(TreeNode* root){
if (root == NULL)
return;
if (root->left != NULL)
{
recursive_traversal(root->left);
}
if (pre != NULL && pre->val>root->val)
{
if (mistake1 == NULL)
{
mistake1 = pre;
mistake2 = root;
}
else
{
mistake2 = root;
}
}
pre = root;
if (root->right != NULL)
{
recursive_traversal(root->right);
}
}
};