Leetcode#669. 修剪二叉搜索树,C++实现

最新推荐文章于 2024-08-23 17:42:22 发布

认真的虎

最新推荐文章于 2024-08-23 17:42:22 发布

阅读量253

点赞数

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/liu502617169/article/details/88894540

版权

数据结构与算法专栏收录该内容

45 篇文章 2 订阅

订阅专栏

1. 题目

在这里插入图片描述

2. 方法一递归

2.1. 代码

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if(root==NULL) return root;
        if((root->val)>R) return trimBST(root->left,L,R);
        if((root->val)<L) return trimBST(root->right,L,R);
        root->left=trimBST(root->left,L,R);
        root->right=trimBST(root->right,L,R);
        return root;
        
    }
    
};

2.2. 结果

在这里插入图片描述

3. 方法一非递归

3.1. 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        queue<TreeNode*> temp;
        while(root!=NULL&&((root->val)<L||(root->val)>R)){
            if(root->val<L){ 
                root=root->right;
                continue;
            }
            if(root->val>R) root=root->left;
        }
        if(root==NULL) return root;
        temp.push(root);
        TreeNode* curr;
        while(!temp.empty()){
            curr=temp.front();
            temp.pop();
            
            while(curr->left!=NULL&&((curr->left->val)<L||(curr->left->val)>R)){
                if((curr->left->val)<L) {
                    curr->left=curr->left->right;
                    continue;
                }
                if((curr->left->val)>R) {
                    curr->left=curr->left->left;
                    continue;
                }
            }
            if(curr->left!=NULL) temp.push(curr->left);
            
            while(curr->right!=NULL&&((curr->right->val)<L||(curr->right->val)>R)){
                if((curr->right->val)<L) {
                    curr->right=curr->right->right;
                    continue;
                }
                if((curr->right->val)>R) {
                    curr->right=curr->right->left;
                    continue;
                }
            }
            if(curr->right!=NULL) temp.push(curr->right);
        }
    return root;  
    }
};