本例中树结构、节点权如下图所示
如何删除节点、子树
本例实现逻辑为直接删除节点及其子节点,未处理存在有左右子节点并需移动逻辑,故将标题命名为为简单删除篇
- 存在左节点或者右节点,删除后需要对子节点移动将在复杂删除篇中更新
- 同时存在左右子节点,不能简单的删除,但是可以通过和后继节点交换后转换为前两种情况将在复杂删除篇中更新
/** * 删除子树 */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)}
运行结果
简答删除逻辑篇到此完结,复杂删除逻辑篇完善中!欢迎关注本人继续跟进技术干货的更新!