当Kotlin遇见数据结构丨实现链式存储的二叉树中删除子树（直接删除篇）

本例中树结构、节点权如下图所示

---

删除节点、子树代码

本例实现逻辑为直接删除节点及其子节点，未处理存在有左右子节点并需移动逻辑，故将标题命名为为直接删除篇

• 存在左节点或者右节点，删除后需要对子节点移动将在善后删除篇中更新
• 同时存在左右子节点，不能简单的删除，但是可以通过和后继节点交换后转换为前两种情况将在善后删除篇中更新

代码语言：javascript复制

    /**
     * 删除节点、子树
     */
    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)

    }

---

运行结果

---

直接删除逻辑篇到此完结，善后删除逻辑篇完善中！欢迎关注本人继续跟进技术干货的更新！

kotlin

0 人点赞