leetcode还原二叉搜索树

最新推荐文章于 2025-06-05 01:30:00 发布

原创最新推荐文章于 2025-06-05 01:30:00 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

C++算法系列同时被 3 个专栏收录

256 篇文章

订阅专栏

设计模式

149 篇文章

订阅专栏

C 算法系列

55 篇文章

订阅专栏

本文介绍了一种修复二叉搜索树中元素错误排序的方法，通过遍历树并使用交换操作来纠正错误，确保树的性质得以恢复。代码示例展示了创建、显示二叉树以及执行修复过程的完整流程。

#include <stdio.h>
#include <iostream>
#include <algorithm>
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {
    }
};
TreeNode *root;
class Solution {
public:
    void create_bt(TreeNode *&root) {
        int e;
        std::cin >> e;
        if (e != -1) {
            root = (TreeNode *)malloc(sizeof(TreeNode));
            root->val = e;
            root->left = nullptr;
            root->right = nullptr;
            create_bt(root->left);
            create_bt(root->right);
        }
    }
    void show_bt(TreeNode *root) {
        if (!root) {
            return;
        }
        std::cout << root->val << std::endl;
        show_bt(root->left);
        show_bt(root->right);
    }
    void transform_tree(TreeNode *root) {
        if (!root) {
            return;
        }
        transform_tree(root->left);
        if (pre_node && pre_node->val > root->val) {
            std::swap(pre_node->val, root->val);
            change = true;
            return;

        }
        pre_node = root;
        transform_tree(root->right);
    }
    void recoverTree(TreeNode *root) {
        do {
            pre_node = nullptr;
            change = false;
            transform_tree(root);
        } while (change);
    }
private:
    TreeNode *pre_node;
    bool change;
};
int main() {
    Solution s;
    s.create_bt(root);
    s.show_bt(root);
    s.recoverTree(root);
    s.show_bt(root);
    
    return 0;
}