算法-递归算法(js树形结构)

大家可能对递归算法并不陌生，但是在我们实战项目中遇到需要使用递归算法的时候不知道如何下手。下面是我个人总结出来的经验，如果有不对的地方欢迎大家指正。

递归算法使用步骤：

　　第一步

　　　　分析目标数据规律：大多数记录的pid都对应了某记录的ID

　　第二步

　　　　分析预期结果：预期结果是否可以通过分析出来的规则直接或间接实现

　　第三步

　　　　确定每次递归的参数和递归终止条件

下面我以常见的多级分类（js树形结构）为例：

const _ = require('lodash')
let a = [{
    id: 1,
    pid: 0
}, {
    id: 2,
    pid: 1
}, {
    id: 3,
    pid: 2
}, {
    id: 4,
    pid: 1
}, {
    id: 5,
    pid: 2
}, {
    id: 6,
    pid: 4
}]
/**
 * 递归使用：
 * 前提：是否有规律
 * 第一步分析目标数据规律：
 *  规律：大多数记录的pid都对应了某记录的ID
 * 第二步分析预期结果：
 *  预期结果：{id:xx,pid:xx,children:[{id:xx,pid:x, ...},...]}
 *  预期结果是否可以通过分析出来的规则直接或间接实现
 * 第三步确定每次递归的参数和递归终止条件
 */
let aHandle = _.groupBy(a, 'pid')  // 运行结果：{"0":[{"id":1,"pid":0}],"1":[{"id":2,"pid":1},{"id":4,"pid":1}],"2":[{"id":3,"pid":2},{"id":5,"pid":2}],"4":[{"id":6,"pid":4}]}

let b = formatTree(aHandle, '0', 'id')
console.log(JSON.stringify(b))

function formatTree(items, parentId, cKey) {
    let result: any[] = []
    if (!items[parentId]) {
        return result
    }
    for (let t of items[parentId]) {
        const temp = formatTree(items, t[cKey], cKey)
        if (temp.length > 0) {
            t.children = temp
        }
        result.push(t)
    }
    return result
}

运行结果如下：

[
    {
        "id":1,
        "pid":0,
        "children":[
            {
                "id":2,
                "pid":1,
                "children":[
                    {
                        "id":3,
                        "pid":2
                    },
                    {
                        "id":5,
                        "pid":2
                    }
                ]
            },
            {
                "id":4,
                "pid":1,
                "children":[
                    {
                        "id":6,
                        "pid":4
                    }
                ]
            }
        ]
    }
]