一次循環(huán)構(gòu)建所有的數(shù)樹, 然后把根節(jié)點找到push進去就好了

let arr = [...]
let sidMap = {}
let map= {} 
for(let i=0 ;i<arr.length;i++){
    let pid = arr[i].source
    let sid = arr[i].target
    if(!map[sid]){
        map[sid]={id:sid, children:[]}
    }
    map[sid].weight = arr[i].weight
    if(!map[pid]){
        map[pid]={id:pid, children:[]}
    }
    map[pid].children.push(map[sid])
    sidMap[sid]=true
}
let ans = []
for(let i=0 ;i<arr.length;i++){
    let pid = arr[i].source
    if(!sidMap[pid]){
        sidMap[pid] = true
        ans.push(map[pid])
    }
}
console.log(ans)

兩次 for循環(huán)
設(shè)置一個 對象ob
第一次 把相同 source 的 對象都 放在一起
及 {2: [{...},{...}],}
第二次 for 就可以 根據(jù)上面的 對象 來 進行操作
具體代碼不貼了

const arrays = [
  {source: '001', target: '002', weight: 10},
  {source: '001', target: '003', weight: 15},
  {source: '002', target: '004', weight: 12},
  {source: '004', target: '005', weight: 10}
]

let tree = [];


const genTree = (tree = [], arr) => {
  if (tree.length === 0) {
    return tree.concat(arr);
  }
  if (tree.some(t => t.target === arr.source)) {
    return tree.map(t => {
      if(t.target === arr.source) {
        if (t.children) {
          t.children.concat([arr]);
        } else {
          t.children = [arr];
        }
      }
      return t;
    });
  } else {
    if(tree.some(t => t.hasOwnProperty('children'))) {
      return tree.map(t => {
        if (t.hasOwnProperty('children')) {
          t.children.concat(genTree(t.children, arr));
        }
        return t;
      });
    }
    return tree.concat(arr);
  }
};


arrays.forEach(arr => {
  tree = genTree(tree, arr);
  // console.log(tree)
})

console.log(tree);

結(jié)果：