leetcode 99. 恢复二叉搜索树（C++）

最新推荐文章于 2022-05-28 01:34:52 发布

nanashi_F

最新推荐文章于 2022-05-28 01:34:52 发布

阅读量389

点赞数

分类专栏： C++ leetcode 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/nanashi_F/article/details/105078855

版权

leetcode 同时被 2 个专栏收录

67 篇文章 1 订阅

订阅专栏

C++

30 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何通过中序遍历来解决LeetCode上的99题——恢复二叉搜索树。讨论了如何找到交换的节点，提出当节点值小于前置节点值时，表明该节点参与了交换，并提供了相应的C++代码实现。

摘要由CSDN通过智能技术生成

题目

二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下，恢复这棵树。

示例 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) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗？

来源：99. 恢复二叉搜索树 - 力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
算法来自官方题解——>恢复二叉搜索树 - 恢复二叉搜索树 - 力扣（LeetCode）

中序遍历法

若要找到交换的节点，就记录中序遍历中的最后一个节点 pred（即当前节点的前置节点），并与当前节点的值进行比较。如果当前节点的值小于前置节点 pred 的值，说明该节点是交换节点之一。
交换的节点只有两个，因此在确定了第二个交换节点以后，可以终止遍历。

代码如下：

class Solution {
public:
    void recoverTree(TreeNode* root) {
        if(!root)   return ;
        vector<TreeNode*> stack;
        TreeNode* x = NULL;
        TreeNode* y = NULL;
        TreeNode* pred = NULL;
        while(!stack.empty() || root){
            while(root){
                stack.push_back(root);
                root = root->left;
            }
            root = stack.back();
            stack.pop_back();
            if(pred && root->val < pred->val){
                y = root;
                if(!x)  x = pred;
                else    break;
            }
            pred = root;
            root = root->right;
        }
        int temp = x->val;
        x->val = y->val;
        y->val = temp;
    }
};