标签:
Two elements of a binary search tree (BST) are swapped by mistake.
Recover the tree without changing its structure.
Note:
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换过来就可以恢复。 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); } } };
LeetCode 99:Recover Binary Search Tree
标签:
原文地址:http://blog.csdn.net/geekmanong/article/details/50419323