算法-遞歸算法(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
                    }
                ]
            }
        ]
    }
]