【LeetCode】99. 恢复二叉搜索树结题报告 (C++)

最新推荐文章于 2024-05-17 20:37:49 发布

置顶暮雨凉初透

最新推荐文章于 2024-05-17 20:37:49 发布

阅读量563

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_32805671/article/details/82629999

本文介绍了一种解决LeetCode上二叉搜索树节点错误交换问题的方法。通过中序遍历找到并交换错误节点，恢复二叉搜索树的正确顺序。提供了一个具体的C++实现示例。

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

原题地址：https://leetcode-cn.com/problems/recover-binary-search-tree/description/

题目描述：

二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下，恢复这棵树。

示例 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) 空间复杂度的解法很容易实现。

解题方案：

对二叉树进行中序遍历，预期的结果为从小到大排序的数列

参考地址：https://blog.youkuaiyun.com/a45872055555/article/details/38685035

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void recoverTree(TreeNode *root) {
        if(root == NULL)
            return ;
        InOrderTraverse(root);
    }
    void InOrderTraverse(TreeNode *root){
        stack<TreeNode *> S;
        TreeNode *p = root;
        vector<int> num;
        TreeNode *tmp = NULL;
        TreeNode *big = NULL;
        TreeNode *small = NULL;
        bool flag = false;
        while(p || !S.empty()){
            if(p){
                S.push(p);
                p = p->left;
            }else{
                p = S.top();
                S.pop();
                if(tmp == NULL){
                    tmp = p;
                    p = p->right;
                    continue;
                }
                if(!flag){
                    if(tmp->val > p->val){
                        big = tmp;
                        small = p;
                        flag = true;
                    }
                }else{
                    if(tmp->val > p->val){
                        small = p;
                        break;
                    }
                }
                tmp = p;
                p = p->right;
            }
        }
        int a = big->val;
        big->val = small->val;
        small->val = a;
    }
};