题目链接:https://leetcode-cn.com/problems/recover-binary-search-tree/
题目描述
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 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
进阶:
- 使用 O(n) 空间复杂度的解法很容易实现。
- 你能想出一个只使用常数空间的解决方案吗?
思路
对正确的二叉搜索树进行中序遍历,那么节点系列将会一直升序;如果两个节点的位置错了,一定会出现降序。
(1)如果在中序遍历时节点值出现了两次降序,第一个错误的节点为第一次降序时较大的节点,第二个错误的节点为第二次降序时较小的节点
(1)如果只出现一次降序,则就是这两个节点
采用中序遍历就可以完成题目要求,递归/非递归/Morris遍历都可以。以下采用非递归方式。
复杂度分析
时间复杂度:O(n)
空间复杂度:O(h)
代码
class Solution {
public:
void recoverTree(TreeNode* root) {
if(!root) return;
stack<TreeNode*> st;
TreeNode *pPrev = nullptr;
TreeNode *err1 = nullptr, *err2 = nullptr;
// 非递归中序遍历
while(!st.empty() || root){
while(root){
st.push(root);
root = root->left;
}
root = st.top();
st.pop();
if(pPrev && pPrev->val > root->val){
err1 = err1 ? err1: pPrev;
err2 = root;
}
pPrev = root;
root = root->right;
}
swap(err1->val, err2->val);
}
};