Recover Binary Search Tree（C++恢复二叉搜索树）

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

coolsunxu

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

阅读量108

点赞数

分类专栏： C++ LeetCode 文章标签：二叉树递归中序遍历恢复有序节点值

本文链接：https://blog.csdn.net/coolsunxu/article/details/118491294

版权

C++ 同时被 2 个专栏收录

735 篇文章 18 订阅

订阅专栏

LeetCode

459 篇文章 1 订阅

订阅专栏

（1）递归+性质

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    vector<int> v;
public:
    void inorder(TreeNode *root) {
        if(!root) return;
        inorder(root->left);
        v.push_back(root->val);
        inorder(root->right);
        return;
    }
    
    vector<int> find_num() {
        vector<int> vec;
        for(int i=0;i<v.size()-1;i++) {
            if(v[i]>v[i+1]) {
                if(vec.size()==0) vec.push_back(i);
                else vec.push_back(i+1);
            }
        }
        return vec;
    }
    
    void helper(vector<int> &vec,TreeNode *root) {
        if(!root) return;
        if(vec.size()==1) {
            if(v[vec[0]]==root->val) root->val=v[vec[0]+1];
            else if(v[vec[0]+1]==root->val) root->val=v[vec[0]];
        } else {
            if(v[vec[0]]==root->val) root->val=v[vec[1]];
            else if(v[vec[1]]==root->val) root->val=v[vec[0]];
        }
        
        helper(vec,root->left);
        helper(vec,root->right);
        return;
    }
    
    void recoverTree(TreeNode* root) {
        inorder(root);
        vector<int> vec=find_num();
        helper(vec,root);
        return;
    }
};