本例中树结构、节点权如下图所示
![](https://img.haomeiwen.com/i9791228/3cd4167888170648.png)
删除节点、子树代码
本例实现逻辑为直接删除节点及其子节点,未处理存在有左右子节点并需移动逻辑,故将标题命名为为直接删除篇
-
存在左节点或者右节点,删除后需要对子节点移动将在善后删除篇中更新
-
同时存在左右子节点,不能简单的删除,但是可以通过和后继节点交换后转换为前两种情况将在善后删除篇中更新
/**
* 删除节点、子树
*/
fun deleteNode(index: Int) {
var indexNode:TreeNode = this
// 比较左子节点
if(indexNode?.leftNode?.value == index){
indexNode?.leftNode = null
return
}
// 比较右子节点
if(indexNode?.rightNode?.value == index){
indexNode?.rightNode = null
return
}
// 递归检查并删除左子节点
leftNode?.deleteNode(index)
// 递归检查并删除右子节点
leftNode?.deleteNode(index)
}
运行结果
![](https://img.haomeiwen.com/i9791228/723da812f31e6b9e.png)
直接删除逻辑篇到此完结,善后删除逻辑篇完善中!欢迎关注本人继续跟进技术干货的更新!
![](https://img.haomeiwen.com/i9791228/b4c392f1912735a8.jpg)
网友评论