代码随想录算法训练营第二十一天|二叉树 part8

669. 修剪二叉搜索树

其实就是判断当前节点的值与搜索区间的关系。

  1. 位于 [ l o w , h i g h ] [low, high] [low,high]: 继续遍历修剪左子树和右子树
  2. 位于 ( − ∞ , l o w ) (-\infty, low) (−∞,low): 返回修剪的右子树
  3. 位于 ( h i g h , + ∞ ) (high, +\infty) (high,+∞): 返回修剪的左子树
cpp 复制代码
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == nullptr)
            return nullptr;
        if (root->val < low)
            return trimBST(root->right, low, high);
        else if (root->val > high)
            return trimBST(root->left, low, high);
        else {
            root->left = trimBST(root->left, low, high);
            root->right = trimBST(root->right, low, high);
        }
        return root;
    }
};

108.将有序数组转换为二叉搜索树

本质就是寻找分割点,分割点作为当前节点,然后递归左区间和右区间。

cpp 复制代码
class Solution {
public:
    TreeNode* construct(vector<int>& nums, int left, int right) {
        //[left, right)
        if (left >= right)
            return nullptr;
        int middle = (left + right)/2;
        TreeNode* root = new TreeNode(nums[middle]);
        //[left, middle)
        root->left = construct(nums, left, middle);
        //[middle + 1, right);
        root->right = construct(nums, middle + 1, right);
        return root;
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return construct(nums, 0, nums.size());
    }
};

538.把二叉搜索树转换为累加树

cpp 复制代码
class Solution {
public:
    TreeNode* prev = nullptr;
    TreeNode* convertBST(TreeNode* root) {
        if (root == nullptr)
            return nullptr;
        
        root->right = convertBST(root->right);
        if (prev != nullptr)
            root->val += prev->val;
        prev = root;
        root->left = convertBST(root->left);
        return root;
    }
};
相关推荐
passer__jw7679 分钟前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾16 分钟前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序24 分钟前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
爱吃生蚝的于勒1 小时前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~1 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^1 小时前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城1 小时前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德2 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz2 小时前
STL--哈希
c++·算法·哈希算法
kingmax542120082 小时前
初三数学,最优解问题
算法