Leetcode 99. 恢复二叉搜索树
题目
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
测试样例
1.
输入: [1,3,null,null,2]
1
/
3
\
2
输出: [3,1,null,null,2]
3
/
1
\
2
2.
输入: [3,1,4,null,null,2]
3
/ \
1 4
/
2
输出: [2,1,4,null,null,3]
2
/ \
1 4
/
3
题解
对于二叉搜索树而言,中序遍历的结果应该是严格递增的。根据题意可以知道,有两个节点被交换了,也就是会有两个位置不会严格递增。第一个地方会发生一次递减,错误节点应该是pre;第二个地方会发生一次递减,而这次是root。当然,这两次可能发生在同一个位置。我们只要分别找到这两个位置,然后交换即可。详细过程见代码
代码
TreeNode *pre=NULL; //记录中序遍历时,前一次遍历的节点
TreeNode *one=NULL; //记录第一次发生错误的错误节点
TreeNode *two=NULL; //记录第二次发生错误的错误节点
bool search(TreeNode *root){
if(root == NULL) return false;
if(search(root->left)) return true;
if(pre != NULL && pre->val>root->val){
if(one==NULL){
one = pre;
two = root;
}else{
two=root;
return true;
}
}
pre = root;
if(search(root->right)) return true;
return false;
}
void recoverTree(TreeNode* root) {
search(root);
swap(one->val,two->val);
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/recover-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。