您的位置:首页 > 文旅 > 美景 > 【二叉树算法题记录】669. 修剪二叉搜索树

【二叉树算法题记录】669. 修剪二叉搜索树

2024/12/23 2:46:16 来源:https://blog.csdn.net/weixin_43627638/article/details/139493319  浏览:    关键词:【二叉树算法题记录】669. 修剪二叉搜索树

题目链接

题目描述

给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。

题目分析

这题很容易忽略的一点是,如果当一个节点值小于给定区间,我们就直接把它的左子树送上去,需要注意的是它的左子树中可能也存在不满足区间的节点。

这题用递归法:

  1. 确定递归参数和返回值:传入当前节点以及边界,返回处理后的节点。
TreeNode* trimBST(TreeNode* root, int low, int high)
  1. 递归终止条件:当遍历到空节点就返回。
if(root == NULL) return NULL;
  1. 单层递归逻辑:
    首先判断当前节点的值,如果小于区间最小值,则只修剪它的右子树(丢弃左子树,因为左子树所有值都比当前根节点小,不满足区间要求),并返回右子树;如果大于区间最大值,则只修剪它的左子树,并返回左子树。如果在区间内,则分别接住修剪后的左子树和右子树即可。
if(root->val < low) {    // 如果当前节点值小于区间TreeNode* right = trimBST(root->right, low, high);  // 修剪它的右子树,并将它的右孩子传给上一层return right;}if(root->val > high) {TreeNode* left = trimBST(root->left, low, high);    // 修剪它的左子树,并将它的左孩子传给上一层return left;}root->left = trimBST(root->left, low, high);root->right = trimBST(root->right, low, high);return root;

整体cpp代码

/*** 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:TreeNode* trimBST(TreeNode* root, int low, int high) {// 终止条件if(root==NULL) return NULL;// 单层递归if(root->val < low) {    // 如果当前节点值小于区间TreeNode* right = trimBST(root->right, low, high);  // 修剪它的右子树,并将它的右孩子传给上一层return right;}if(root->val > high) {TreeNode* left = trimBST(root->left, low, high);    // 修剪它的左子树,并将它的左孩子传给上一层return left;}root->left = trimBST(root->left, low, high);root->right = trimBST(root->right, low, high);return root;}
};

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com