代码随想录算法训练营第22天|● 235. 二叉搜索树的最近公共祖先 ● 701.二叉搜索树中的插入操作 ● 450.删除二叉搜索树中的节点

二叉搜索树的最近公共祖先

题目链接

https://leetcode.cn/problems/lowest-common-ancestor-of-a-binary-search-tree/description/

思路

因为是有序树,所以 如果 中间节点是 q 和 p 的公共祖先,那么 中节点的数组 一定是在 [p, q]区间的。即 中节点 > p && 中节点 < q 或者 中节点 > q && 中节点 < p。所以子需要从上到下遍历第一次遇见符合条件的就是最小的公共祖先

代码:

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
           if(root==null){
             return null;
         }
        if(root.val>p.val&&root.val>q.val){//root.val比p,q的值都大,说明最小祖先在左子树
            TreeNode left=lowestCommonAncestor(root.left,p,q);
            if(left!=null){
                return left;
            }
        }
        if(root.val<p.val&&root.val<q.val){//root.val比p,q的值都小,说明最小祖先在右子树
            TreeNode right=lowestCommonAncestor(root.right,p,q);
            if(right!=null){
                return right;
            }
        }
        return root;
    }
}

二叉搜索树中的插入操作

题目链接

https://leetcode.cn/problems/insert-into-a-binary-search-tree/

思路:

基本情况:如果传入的树根节点root为null,表示树为空。此时,函数创建一个新的TreeNode对象,用val初始化,并返回这个新节点作为新的树根。

递归情况:如果树不为空,函数会根据val和当前节点root的值进行操作:

如果val小于root.val,说明val应该被插入到左子树中。因此,函数会递归调用自身,将val插入到root.left(左子节点),并将更新后的左子树赋值回root.left。

如果val大于root.val,说明val应该被插入到右子树中。同理,函数会递归调用自身,将val插入到root.right(右子节点),并将更新后的右子树赋值回root.right。

返回结果:无论哪种情况,函数最后都会返回当前节点root。这使得递归调用能够逐级更新原树结构,直到插入操作完成并返回最初的树根节点。

代码:

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
             if(root==null){
            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;
    }
}

删除二叉搜索树中的节点

题目链接

https://leetcode.cn/problems/delete-node-in-a-bst/

思路:

首先判断根节点是否为空,若为空则返回null。然后判断根节点的值是否等于要删除的值,若相等则分四种情况处理:若该节点没有左右子节点,则直接返回null;若只有左子节点,则返回左子节点;若只有右子节点,则返回右子节点;若既有左子节点又有右子节点,则找到右子树中最小的节点,将其值替换要删除的节点的值,并删除右子树中最小的节点。接着判断要删除的值是否小于根节点的值,若是则递归地在左子树中删除该值;若大于根节点的值,则递归地在右子树中删除该值。最后返回修改后的根节点。

代码:

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 * int val;
 * TreeNode left;
 * TreeNode right;
 * TreeNode() {}
 * TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) {
 * this.val = val;
 * this.left = left;
 * this.right = right;
 * }
 * }
 */
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
     // 如果根节点为空,直接返回null
        if (root == null) {
            return null;
        }
        // 如果找到需要删除的节点
        if (root.val == key) {
            // 如果该节点没有子节点,直接删除(返回null)
            if(root.left==null&&root.right==null){
                return null;
            }
            // 如果该节点只有左子节点,用左子节点替换该节点后返回
            else if(root.left!=null&&root.right==null){
                return root.left;
            }
            // 如果该节点只有右子节点,用右子节点替换该节点后返回
            else if(root.left==null&&root.right!=null){
                return root.right;
            }
            // 如果该节点既有左子节点又有右子节点,找到右子树中最小的节点,将其值替换该节点的值,然后递归删除右子树中的该最小节点
            else if(root.left!=null&&root.right!=null){
                TreeNode temp=root.right;
                while (temp.left!=null){
                    temp=temp.left;
                }
                temp.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;
    }
}
相关推荐
Once_day8 分钟前
Linux之kernel(1)系统基础理论(6)
linux·操作系统
阿湯哥28 分钟前
ES的预置分词器
linux·服务器·elasticsearch
lisanndesu31 分钟前
贪心算法--
算法·贪心算法
巷95533 分钟前
机器学习篇——决策树基础
算法·决策树·机器学习
猎人everest35 分钟前
机器学习之正则化
人工智能·算法·机器学习
YaoSolar38 分钟前
刷题记录(LeetCode 78 子集)
算法·leetcode
wanhengidc1 小时前
物理服务器的作用都有哪些?
运维·服务器
熊峰峰1 小时前
数据结构第八节:红黑树(初阶)
开发语言·数据结构·c++·算法
Linux运维技术栈1 小时前
ActiveMQ 5.1.3:单节点与集群部署实战指南
运维·架构·activemq
艾思科蓝 AiScholar1 小时前
SCI期刊推荐 | 免版面费 | 计算机领域:信息系统、软件工程、自动化和控制
运维·人工智能·深度学习·信息可视化·自然语言处理·自动化·软件工程