JavaScript实现单层数组转多层树结构

/**
 * 列表转为树，pid值未知。思路：第一遍遍历构造为map，第二遍
 * @param {*} list 
 * @returns 
 */
 export function listToTree(list) {
  let treeData = []
  // 第一次遍历，构造一个map
  let tmpObj = {}
  list.forEach(item => { // 这里也可以把值设置为数组的索引，这样的话在这里可以不必存储这么大空间了
    tmpObj[item.id] = item
  });
  list.forEach(v => {
    const parent = tmpObj[v.parentId]
    // 如果有parentId并且在临时map中存在
    if (v.parentId && parent) {
      // 那么把当前元素追加到目标元素children之内，并且把当前key删除
      if (!parent.children) {
        parent.children = []
      }
      parent.children.push(v)
    } else {
      treeData.push(v)
    }
  })
  // 再遍历，根据pid在map中找是否有key，如果有，说明是上级，那么追加到children中；如果没有，那么就跳过，表示没有上级
  return treeData;
}

let newArray = [];
arr.forEach(res => {
   if(res.parentId === 0) {
        newArray.push(res);
    }
});
this.arrToTree(arr, newArray) ;
console.log(newArray);  // 最终得到的新数组；
function arrToTree(list, arr) {
    arr.forEach(res => {
        list.forEach((ret, index) => {
            if(res.id === ret.parentId) {
                if(!res.hasOwnProperty('children')){
                    res.children = [];
                };
                res.children.push(ret);
            }
        })
        if(res.hasOwnProperty('children')) {
            this.arrToTree(list, res.children);
        }
    })
}