[Leetcode] Recover Binary Search Tree

最新推荐文章于 2018-08-21 10:07:21 发布

leo524891010

最新推荐文章于 2018-08-21 10:07:21 发布

阅读量817

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/algorithmengine/article/details/8300935

Leetcode 专栏收录该内容

135 篇文章

订阅专栏

本文介绍了一种通过中序遍历收集二叉树节点值并排序，再按排序后的顺序重新赋值给原节点的方法，以此来纠正被错误修改的二叉树结构。

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    vector<int> val;
    vector<TreeNode*> seq;

public:
    void recoverTree(TreeNode *root) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        val.clear();
        seq.clear();
        inorderTraverse(root);
        
        sort(val.begin(), val.end());
        for (int i = 0; i < seq.size(); ++i)
        {
            seq[i]->val = val[i];
        }
    }
    
    void inorderTraverse(TreeNode* root)
    {
        if (!root)
            return;
        
        inorderTraverse(root->left);
        val.push_back(root->val);
        seq.push_back(root);
        inorderTraverse(root->right);        
    }
};