Kotlin抽象二叉树类add方法改造及构造器约束实现方案问询

方案1：抽象工厂方法约束节点构造 #

在抽象BinaryTree中定义一个抽象的节点创建方法，强制子类实现节点的实例化逻辑，这样就能把add(value: E)改为非抽象方法，依赖该工厂方法创建节点后调用add(node: N)。

示例代码：

abstract class Node<E>(open val value: E)

abstract class BinaryTree<E, N : Node<E>> {
    // 抽象方法：子类必须实现节点创建逻辑
    protected abstract fun createNode(value: E): N
    
    abstract fun add(node: N)
    
    // 非抽象的add方法，依赖createNode创建节点
    fun add(value: E) {
        val node = createNode(value)
        add(node)
    }
}

// 子类示例：二叉搜索树
class BstNode<E>(override val value: E) : Node<E>(value)

class BinarySearchTree<E : Comparable<E>> : BinaryTree<E, BstNode<E>>() {
    override fun createNode(value: E): BstNode<E> = BstNode(value)
    
    override fun add(node: BstNode<E>) {
        // 二叉搜索树的节点插入逻辑
    }
}

这个方案完全符合面向抽象设计原则，没有额外依赖，代码可读性强，且子类的扩展性不受限制——后续继承BinarySearchTree时，只需重写createNode和add(node)即可。

方案2：通过接口解耦节点构造逻辑 #

定义一个专门的节点构造接口，将构造逻辑从树类中分离，通过依赖注入的方式传入抽象树类，既保证add(value: E)非抽象，又强制子类提供合法的节点构造实现。

示例代码：

interface NodeConstructor<E, N : Node<E>> {
    fun create(value: E): N
}

abstract class Node<E>(open val value: E)

abstract class BinaryTree<E, N : Node<E>>(private val nodeConstructor: NodeConstructor<E, N>) {
    abstract fun add(node: N)
    
    fun add(value: E) {
        val node = nodeConstructor.create(value)
        add(node)
    }
}

// 子类实现
class BstNode<E>(override val value: E) : Node<E>(value)

class BinarySearchTree<E : Comparable<E>> : BinaryTree<E, BstNode<E>>(
    object : NodeConstructor<E, BstNode<E>> {
        override fun create(value: E): BstNode<E> = BstNode(value)
    }
) {
    override fun add(node: BstNode<E>) {
        // 二叉搜索树插入逻辑
    }
}

该方案的优势是解耦了树结构与节点构造，灵活性更高——如果需要更换节点类型，只需传入不同的NodeConstructor实现即可，无需修改树类本身。

方案3：反射+具体化泛型（谨慎使用） #

利用Kotlin的具体化泛型和反射机制，直接在add(value: E)中实例化节点。此方案代码最简洁，但依赖反射，存在运行时安全风险（若节点无匹配构造函数会抛出异常），且性能略低，适合对代码简洁性要求高、能保证节点构造规范的场景。

示例代码：

abstract class Node<E>(open val value: E)

abstract class BinaryTree<E, N : Node<E>> {
    abstract fun add(node: N)
    
    inline fun <reified T : N> add(value: E) {
        // 查找并调用参数为E的构造函数
        val constructor = T::class.constructors.firstOrNull {
            it.parameters.size == 1 && it.parameters[0].type.classifier == E::class
        } ?: error("Node类必须提供参数为$E的构造函数")
        
        val node = constructor.call(value)
        add(node)
    }
}

// 使用示例
class BstNode<E>(override val value: E) : Node<E>(value)

class BinarySearchTree<E : Comparable<E>> : BinaryTree<E, BstNode<E>>() {
    override fun add(node: BstNode<E>) {
        // 插入逻辑
    }
}

// 调用方式
val bst = BinarySearchTree<Int>()
bst.add<BstNode<Int>>(10)

内容的提问来源于stack exchange，提问作者Luke Abruzese