LeetCode - Recover Binary Search Tree

最新推荐文章于 2021-06-13 13:54:56 发布

原创最新推荐文章于 2021-06-13 13:54:56 发布 · 115 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

180 篇文章

订阅专栏

本文介绍了解决二叉搜索树中两个节点值被错误交换的问题，提供了两种不同的算法实现，一种通过排序中间遍历结果进行修正，另一种在遍历过程中直接找到并交换错误的节点。

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

解法一

class Solution {
public:
    void recoverTree(TreeNode* root) {
        vector<TreeNode*> trees;
        vector<int> vals;
        inorder(root, trees, vals);
        sort(vals.begin(), vals.end());
        for(int i=0;i<trees.size();i++){
            if(trees[i]->val!=vals[i]) trees[i]->val=vals[i];
        }
    }
    void inorder(TreeNode* root, vector<TreeNode*>& trees, vector<int>& vals){
        if(!root) return;
        inorder(root->left, trees, vals);
        trees.push_back(root);
        vals.push_back(root->val);
        inorder(root->right, trees, vals);
    }
};

解法二

class Solution {
public:
    void recoverTree(TreeNode* root) {
        TreeNode* first=NULL;
        TreeNode* second=NULL;
        TreeNode* pre = NULL;
        inorder(root, pre, first, second);
        if(first && second) swap(first->val, second->val);
    }
    void inorder(TreeNode* root, TreeNode*& pre, TreeNode*& first, TreeNode*& second){
        if(!root) return;
        inorder(root->left, pre, first, second);
        if(!pre) pre=root;
        else{
            if(pre->val>root->val){
                if(!first) first = pre;
                second = root;
            }
            pre = root;
        }
        inorder(root->right, pre, first, second);
    }
};