一维数组转换成树形结构

原数组

const arr = [{id:4,pid:3},{id:1,pid:22},{id:3,pid:2},{id:2,pid:1}];

一. reduce + 递归

const arrayToTree = (arr, pid) => {
    return arr.reduce((res, current) => {
        if (current['pid'] === pid) {
            current.children = arrayToTree(arr, current['id']);
            return res.concat(current);
        }
        return res;
    }, []);
};

二. forEach

function toTree(data) {
    const res = [];
    const map = {};
    data.forEach((item) => {
        map[item.id] = item
    })
    data.forEach((item) => {
        let parent = map[item.pid];
 
        if(parent) {
            (parent.children || (parent.children = [])).push(item)
        } else {
            res.push(item)
        }
    })
    return res
}

调用： arrayToTree(arr,22)