ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode 99. 恢复二叉搜索树

2020-08-08 21:04:29  阅读:313  来源: 互联网

标签:right TreeNode val int 二叉 99 root LeetCode left


题目描述链接:https://leetcode-cn.com/problems/recover-binary-search-tree/

解题思路:中序遍历二叉树,找出不符合条件的两个节点,然后重新遍历进行恢复

LeetCode C++求解代码:

/**
 * 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 {
public:
    void recoverTree(TreeNode* root) {

        stack<TreeNode *>S;
        TreeNode* p=root;
        TreeNode* q;
        vector<int>res;
        while(!S.empty()||p){
            if(p){
                S.push(p);
                p=p->left;
            }
            else{
               q=S.top();
               S.pop();
               res.push_back(q->val);
               p=q->right;
            }
        }
        int x=-1,y=-1;
        for(int i=0;i<res.size()-1;++i){
            if(res[i+1]<res[i]){
                y=res[i+1];
                if(x==-1){
                    x=res[i];
                }
                else{
                    break;
                }
            }
        }
     recover(root,2,x,y);

    }
      
   void recover(TreeNode *root,int count,int x,int y){
        if(root){
            
            if(root->val==x){

                root->val=y;
                count--;
               
            }
            else  if(root->val==y){

                root->val=x;
                count--;
               
            }
            if(count==0){
                return ;
            }


            recover(root->left,count,x,y);
            recover(root->right,count,x,y);
        }
    }





};

时间复杂度:O(n)   空间复杂度O(n)

标签:right,TreeNode,val,int,二叉,99,root,LeetCode,left
来源: https://www.cnblogs.com/zzw-/p/13460240.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有