JS一维数组转多维数组树的方法

/**
 * 将一维数组转换为树结构
 * @param {Array} data - 输入的一维数组
 * @return {Array} 树结构数组
 */
function arrayToTree(data) {
    // 创建一个id映射，用来快速查找元素
    const idMap = {};
    // 遍历数组，初始化每个元素的children字段，并将其存入idMap
    data.forEach(item => {
        idMap[item.id] = { ...item, children: [] };
    });

    // 初始化一个数组，用来存储根节点
    const tree = [];

    // 再次遍历数组，根据parentId建立树结构
    data.forEach(item => {
        const { id, parentId } = item;
        if (parentId == '' || parentId == null || parentId == undefined) {
            // 如果parentId为null或undefined，表示是根节点
            tree.push(idMap[id]);
        } else {
            // 否则，将当前元素添加到其父元素的children字段中
            if (idMap[parentId]) {
                idMap[parentId].children.push(idMap[id]);
            }
        }
    });

    return tree;
}

let data = [
    { id: '1', parentId: '' },
    { id: '1_1', parentId: '1' },
    { id: '2', parentId: '' }
];

let tree = arrayToTree(data);