14_组合模式

一、结点类

package com.study.composite

interface INode {
    /**
     * 打印当前节点的信息
     */
    fun print()
}

package com.study.composite

/**
 * 树节点
 * nodeName当前结点名称
 * depth当前结点的深度
 */
class BranchNode(private val nodeName: String, private val depth: Int) : INode {
    val childNodes = arrayListOf<INode>()
    /**
     * 添加子节点
     */
    fun addChildNode(node: INode) {
        childNodes.add(node)
    }

    override fun print() {
        for (i in 0..depth) print("-")
        println("树：$nodeName::$depth")
    }
}

package com.study.composite

/**
 * 叶子结点
 */
class LeafNode(val nodeName: String, val depth: Int) : INode {
    override fun print() {
        for (i in 0..depth) print("-")
        println("叶：$nodeName::$depth")
    }
}

二、主程序调用

package com.study.composite

/**
 * 定义和类型
    将对象组合成树形结构以表示“部分-整体”的层次结构
    组合模式使客户端对单个对象和组合对象保持一致的方式处理

    类型：结构型
    适用场景
    希望客户端可以忽略组合对象和单个对象的差异时
    处理一个树型结构时

    优点
    清楚地定义分层次的复杂对象，表示对象的全部或部分层次
    让客户端忽略了层次的差异，方便对整个层次结构进行控制
    简化客户端代码
    符合开闭原则

    缺点
    限制类型时会较为复杂
    使设计变得更加抽象
 */
fun main() {
    val rootNode = BranchNode("中国", 0)

    val branchNode1 = BranchNode("湖北", 1)
    val branchNode2 = BranchNode("湖南", 1)

    val leafNode11 = LeafNode("武汉", 2)
    val leafNode12 = LeafNode("十堰", 2)

    val leafNode21 = LeafNode("长沙", 2)
    val leafNode22 = LeafNode("吉首", 2)

    rootNode.addChildNode(branchNode1)
    rootNode.addChildNode(branchNode2)

    branchNode1.addChildNode(leafNode11)
    branchNode1.addChildNode(leafNode12)

    branchNode2.addChildNode(leafNode21)
    branchNode2.addChildNode(leafNode22)

    printNode(rootNode)
}

/**
 * 遍历节点
 */
fun printNode(node: INode) {
    node.print()
    //树结点
    if (node is BranchNode) {
        for (tempNode in node.childNodes)
            printNode(tempNode)
    }
}

三、运行结果