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

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/qq_15609303/article/details/102297139

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

---

删除节点、子树代码

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

• 存在左节点或者右节点，删除后需要对子节点移动将在善后删除篇中更新

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

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

    }

---

运行结果

---