Recover Binary Search Tree

最新推荐文章于 2018-08-21 10:07:21 发布

angelewings

最新推荐文章于 2018-08-21 10:07:21 发布

阅读量417

点赞数

本文链接：https://blog.youkuaiyun.com/angelewings/article/details/40453031

版权

LeetCode 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种在O(n)时间复杂度内修复被错误修改节点值的二叉搜索树的方法。通过两种实现方式——使用栈进行迭代遍历和递归遍历，找到并交换两个错误节点的位置，使树恢复有序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

solution using O(n) space，二叉树中序遍历

class Solution {
public:
void recoverTree(TreeNode *root) {
if(root==NULL)
return;
TreeNode *first=NULL,*second=NULL,*last=NULL;
stack<TreeNode*> stack;
while(root){
stack.push(root);
root=root->left;
}

while(!stack.empty()){
TreeNode *p=stack.top();
stack.pop();

if(last==NULL)
last=p;
else if(p->val<last->val){
if(first==NULL){
first=last;
second=p;
}
else{
second=p;
break;
}
}

last=p;

p=p->right;
while(p){
stack.push(p);
p=p->left;
}
}

if(first){
int temp=first->val;
first->val=second->val;
second->val=temp;
}
}
};

constant space solution，直接用递归就不用花n的空间了

class Solution {
TreeNode *first,*second,*last;
public:
void recover(TreeNode *root){
if(root==NULL)
return;
recover(root->left);

if(last!=NULL&&last->val>root->val){
if(first==NULL){
first=last;
second=root;
}
else
second=root;
}

last=root;
recover(root->right);
}
void recoverTree(TreeNode *root) {
if(root==NULL)
return;
first=second=last=NULL;
recover(root);

if(first){
first->val^=second->val;
second->val^=first->val;
first->val^=second->val;
}
}
};