99. 恢复二叉搜索树

Python ml

于 2024-02-16 00:17:52 发布

阅读量105

点赞数 2

分类专栏： LeetCode 文章标签：数据结构算法

本文链接：https://blog.csdn.net/m0_48711099/article/details/136124940

版权

LeetCode 专栏收录该内容

180 篇文章 0 订阅

订阅专栏

文章介绍了如何使用递归方法inorder遍历二叉树并将节点值进行调整，使得在中序遍历过程中，相邻节点值满足升序。recoverTree函数用于执行这个操作。

摘要由CSDN通过智能技术生成

class Solution {
public:
    void inorder(TreeNode* root, vector<TreeNode*>& nodes) {
        if (root == nullptr) 
            return;
        inorder(root->left, nodes);
        nodes.push_back(root);
        inorder(root->right, nodes);
    }
    void recoverTree(TreeNode* root) {
        vector<TreeNode*>nodes;
        inorder(root,nodes);
        int idx1=-1,idx2=-1;
        for(int i=0;i<nodes.size()-1;++i){
            if(nodes[i]->val>=nodes[i+1]->val){
                if(idx1==-1) {
                    idx1=i;
                    idx2=i+1;  //可能错误的是相邻节点
                }
                else idx2=i+1;
            }
        }
        swap(nodes[idx1]->val,nodes[idx2]->val);
    }
};