Kotlin 树状结构的遍历 & 递归构建一棵树源代码实例

 

package com.ak47.cms.cms.tree

import com.ak47.cms.cms.vo.TreeVO

object TreeUtil {

    /**
     * 递归遍历树
     * @param node 当前树节点
     * @param visitor 游历函数
     */
    fun visitTree(node: TreeVO, visitor: (TreeVO) -> Unit) {
        // 调用游历函数
        visitor(node)

        node.children.forEach {
            visitTree(it, visitor)
        }
    }


    /**
     * 根据节点列表,递归构建一棵树
     * @param nodes 节点列表
     * @param category 树的类型
     */
    fun buildTree(nodes: List<TreeVO>, category: TreeCategory): TreeVO {
        val treeNodes = mutableListOf<TreeVO>()

        val rootNodes = nodes.filter { it.parentCode == "-1" && it.category == category.name }
        rootNodes.forEach {
            buildChildren(it, nodes)
            t