二叉树篇--代码随想录算法训练营第十九天|701.二叉搜索树中的插入操作,450.删除二叉搜索树中的节点,669. 修剪二叉搜索树,108.将有序数组转换为二叉搜索树,538.把二叉搜索树转换为累加树

701.二叉搜索树中的插入操作

题目链接:. - 力扣(LeetCode)

讲解视频:原来这么简单? | LeetCode:701.二叉搜索树中的插入操作

题目描述:

给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。

注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

示例 1:

输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:

解题思路:

根据该题目可判断--只在叶子节点处执行插入操作。

只需根据二叉搜索树的性质遍历至空节点后,插入新节点即可。

代码: 

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if(root == nullptr) 
        {
            TreeNode* node = new TreeNode(val);
            return node;
        }
        if(root->val > val) root->left = insertIntoBST(root->left,val); 
        if(root->val < val) root->right = insertIntoBST(root->right,val);
        
        return root;
    }
};

450.删除二叉搜索树中的节点

题目链接:. - 力扣(LeetCode)

讲解视频:调整二叉树的结构最难!| LeetCode:450.删除二叉搜索树中的节点

题目描述:

给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。

一般来说,删除节点可分为两个步骤:

  1. 首先找到需要删除的节点;
  2. 如果找到了,删除它。

示例 1:

输入:root = [5,3,6,2,4,null,7], key = 3
输出:[5,4,6,2,null,null,7]
解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。

解题思路:

本题麻烦之处在于删除节点后考虑情况多,分为如下5种:

  1. 要删除的节点不存在。--返回空指针
  2. 删除节点左右子树均为空(叶子)。--返回空指针
  3. 删除节点左子树空,右子树不空。--返回右子树中心节点
  4. 删除节点右子树空,左子树不空。--返回左子树中心节点
  5. 删除节点左右子树均不空。--根据搜索树性质,当前要删除的节点左子树可以并入其右子树中的左子树最左端,如下图所示。返回右子树中心节点

代码: 

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root == nullptr) return nullptr;
        if(root->val == key)
        {
            if(root->left && !root->right) return root->left;
            if(!root->left && root->right) return root->right;
            if(!root->left && !root->right) return nullptr;
            if(root->left && root->right) 
            {
                TreeNode* cur = root->right;
                while(cur->left) cur = cur->left;
                cur->left = root->left;
                return root->right;
            }
        }
        
        if(root->val > key) root->left = deleteNode(root->left,key);
        if(root->val < key) root->right = deleteNode(root->right,key);

        return root;
    }
};

669. 修剪二叉搜索树

题目链接:. - 力扣(LeetCode)

讲解视频:你修剪的方式不对,我来给你纠正一下!| LeetCode:669. 修剪二叉搜索树

题目描述:

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

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

示例 1:

输入:root = [1,0,2], low = 1, high = 2
输出:[1,null,2]

解题思路:

存在如下两种删除情况:

  1. 当前节点数小于左边界。
  2. 当前节点数大于右边界。

根据二叉搜索树性质,解决方式如下:

  1. 当前节点数小于左边界-->当前节点及其左子树均删除,只需对其右子树进行递归查找
  2. 当前节点数大于右边界-->当前节点及其右子树均删除,只需对其左子树进行递归查找

代码: 

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);
        if(root->val > high) return trimBST(root->left,low,high);
        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }
};

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

题目链接:. - 力扣(LeetCode)

讲解视频:构造平衡二叉搜索树!| LeetCode:108.将有序数组转换为二叉搜索树

题目描述:

给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 

平衡

 二叉搜索树。

示例 1:

输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:

解题思路:

 有序数组+左右子树高度差绝对值不超过1(平衡树)-->使用二分查找来分割左右子树元素

代码: 

class Solution {
public:
    TreeNode* createSearchTree(vector<int>& nums, int left, int right)
    {
        if(left > right) return nullptr;
        int mid = left + (right - left) / 2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = createSearchTree(nums,left,mid-1);
        root->right = createSearchTree(nums,mid+1,right);
        return root;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        int left = 0, right = nums.size()-1;
        TreeNode* root = createSearchTree(nums,left,right);
        return root;
    }
};

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

题目链接:. - 力扣(LeetCode)

讲解视频:普大喜奔!二叉树章节已全部更完啦!| LeetCode:538.把二叉搜索树转换为累加树

题目描述:

给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下,二叉搜索树满足下列约束条件:

  • 节点的左子树仅包含键 小于 节点键的节点。
  • 节点的右子树仅包含键 大于 节点键的节点。
  • 左右子树也必须是二叉搜索树。

注意:本题和 1038: . - 力扣(LeetCode) 相同

示例 1:

输入:[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出:[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

解题思路:

可以看出树中累加的顺序是右中左,所以我们需要反中序遍历这个二叉树,然后顺序累加就可以了。

使用双指针法--pre和cur。解法同530.二叉搜索树的最小绝对差

代码: 

class Solution {
public:
    TreeNode* pre = nullptr;
    TreeNode* convertBST(TreeNode* root) {
        if(root == nullptr) return nullptr;
        convertBST(root->right);
        if(pre)
        {
            root->val += pre->val;
        }
        pre = root;
        convertBST(root->left);
        return root;
    }
};

二叉树篇结束总结

  • 9
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值