前言:在学习BST时,发现查找插入等操作都很好理解和实现,而删除节点的操作情况比较复杂,故通过自己的理解整理如下,本文适合初学者理解并自己动手实现BST删除操作。
在很多文章中提到,删除节点可以考虑以下三种情况:
- 没有左右子节点
- 存在左节点或者右节点
- 同时存在左右子节点
这种思路是最简单的思路很好理解,但是实现起来会发现代码有点冗杂。在本文中,将按照以下思路来进行讲解:
- 没有左子树(包括了没有左子树有右子树,没有左子树也没有右子树)
- 有左子树但没有右子树
- 既有左子树,又有右子树
首先,我们来想一下删除一个单链表的节点该如何操作?(1)找到要删除的节点,(2)记录该节点的前一个节点(3)将前一节点链接到该节点的后一节点。这样就将该节点所隔离,根据java的垃圾回收机制,该节点将被删除回收内存。
那么如果二叉树就像一个单链表一样,那不是就很好解决了?所以删除二叉树节点的基本情况就是要删除的节点没有左子树,或者没有右子树(current.left==null || current.right==null),这时少了一路分支,处理起来就像单链表一样,分为三步:(1)找到要删除的节点(2)记录该节点的父节点(3)将父节点链接到该节点的后一节点。但是在执行第三步的时候要注意:父节点有left和right两个指针,需要判断是哪个指针,也就是current是父节点的左子树还是右子树;该节点的后一节点就是那个唯一的不为空的节点。
情况一:没有左子树
情况二:有左子树没有右子树
下图分别给出了要删除的节点只有右子树,只有左子树,左右子树都没有(叶子节点)的情况。