LeetCode每日1题--删除二叉搜索树中的节点

72 阅读3分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第16天,点击查看活动详情

前言

算法的重要性不言而喻!区分度高!

现在学习的门槛低了,只有能上网每个人都可以学编程!培训班6个月就可以培养出来能干活的人,你怎么从这些人中脱颖而出?没错!就是学算法,学一些底层和基础的东西。

说的功利点是为了竞争,卷死对手。真心话说就是能提高自己的基础能力,为技术可持续发展做好充分的准备!!!

提前入门学习书籍:CPrimerPlus、大话数据结构

image-20220705103735001

刷题网站

代码随想录 (programmercarl.com)

leetcode

我是按照代码随想录提供的刷题顺序进行刷题的,大家也可以去刷leetcode最热200道,都可以

刷题嘛,最重要的就是坚持了!!!

画图软件

OneNote

这个要经常用,遇见不懂的流程的话就拿它画一画!

笔记软件

Typoral

题目

力扣题目链接

解析

递归三部曲

  1. 确定递归的参数和返回值

这里用题目给出的函数和返回值就可以,但是为了美观我们在写一个函数在下面


public TreeNode delete(TreeNode root,int key){}

  1. 确定递归终止的条件

递归的终止条件一般我们就用根节点为null的话那就返回null

if(root == null ) return null;
  1. 确定单层递归的逻辑

这里是删除二叉搜索树中的关键,也是一个难点。因为我们不仅要删除节点,还要对二叉搜索树的结构进行变化!

但是经过我们的仔细分析,有什么难做的呢?

删除的逻辑分为5种,如下:

  1. 没找到被删除的节点,那就直接返回了
  2. 找到节点了,又分为4种情况
  • 被删除节点的左右子树为空,删除节点返回null为根节点
  • 被删除节点的左子树为空,右子树不为空,删除节点,右孩子直接补位返回右子树为根节点
  • 被删除节点的右子树为空,左子树不为空,删除节点,左孩子直接补位返回左子树为根节点
  • 被删除节点的左右右子树为都不为空,找到要删除节点右子树的左节点,将要删除节点的左子树赋值给它

被删除节点左右子树都不会空的情况过程如图:

008eGmZEly1gnbj3k596mg30dq0aigyz.gif

if(root.val < key){
    root.right = delete(root.right,key);
}else if(root.val > key){
    root.left = delete(root.left,key);
}else{
    if(root.left == null) return root.right;
    if(root.right == null) return root.left;
    TreeNode cur = root.right;
    //找到要删除节点右子树的左节点
    while(cur.left != null){
        cur = cur.left;
    }
    cur.left = root.left;
    root = root.right;
}

完整代码

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        root = delete(root,key);
        return root;
    }

    public TreeNode delete(TreeNode root,int key){
        if(root == null ) return null;

        if(root.val < key){
            root.right = delete(root.right,key);
        }else if(root.val > key){
            root.left = delete(root.left,key);
        }else{
            if(root.left == null) return root.right;
            if(root.right == null) return root.left;
            TreeNode cur = root.right;
            //找到要删除节点右子树的左节点
            while(cur.left != null){
                cur = cur.left;
            }
            cur.left = root.left;
            root = root.right;
        }
        return root;
    }
}