如何將JavaScript將數(shù)組轉為樹形結構

更新時間：2021年06月02日 09:56:48 作者：蘭村_web

我們經(jīng)常會碰到樹形數(shù)據(jù)結構，比方組織層級、省市縣或者者動植物分類等等數(shù)據(jù)，那么如何將JavaScript將數(shù)組轉為樹形結構，本文就詳細的來了解一下

1.需求

后臺給了一個這樣的數(shù)據(jù)讓咱前端去轉換為樹形結構（沒有重復數(shù)據(jù)）。不多說，先來看看給了一個怎樣的數(shù)組數(shù)據(jù)，轉換為怎樣的樹形結構。

服務器傳過來的數(shù)組

const arr = [
      [
        {"deptId":"D019",
        "deptName":"銷售部"},
        {"deptId":"D019101",
        "deptName":"華北銷售中心"}
      ],[
        {"deptId":"D083",
        "deptName":"音樂事業(yè)部"}
      ],[
        {"deptId":"D027",
        "deptName":"杭州研究院"},
        {"deptId":"D027048",
        "deptName":"技術工程事業(yè)部"},
        {"deptId":"D027048002",
        "deptName":"項目管理中心"}
      ],[
        {"deptId":"D027",
        "deptName":"杭州研究院"},
        {"deptId":"D027048",
        "deptName":"技術工程事業(yè)部"}
      ],[
        {"deptId":"D027",
        "deptName":"杭州研究院"},
        {"deptId":"D027048",
        "deptName":"技術工程事業(yè)部"}
      ]
    ]

最終轉換為

const arr = [
    {
      deptId: 'D019',
      deptName: '銷售部',
      children: [{
        deptId: 'D019101',
        deptName: '華北銷售中心',
        children: [],
      }]
    },
    {
      deptId: 'D083',
      deptName: '音樂事業(yè)部',
      children: []
    },
    {
      deptId: 'D027',
      deptName: '杭州研究院',
      children: [{
        deptId: 'D027048',
        deptName: '技術工程事業(yè)部',
        children: [{
          deptId: 'D027048002',
          deptName: '項目管理中心',
          children: []
        }]
      }]
    },
  ]

2.上代碼（在reactHooks中開發(fā)）

const [treeData, setTreeData] = useState([]);
  console.log(treeData); //treeData為最終需要的樹形結構，（在我本地瀏覽器打印出來是正確的）
  
  useEffect(() => {
    const str = '[[{"deptId":"D019","deptName":"銷售部"},{"deptId":"D019101","deptName":"華北銷售中心"}],[{"deptId":"D019","deptName":"銷售部"},{"deptId":"D019101","deptName":"華北銷售中心"}],[{"deptId":"D083","deptName":"音樂事業(yè)部"}],[{"deptId":"D027","deptName":"杭州研究院"},{"deptId":"D027048","deptName":"技術工程事業(yè)部"},{"deptId":"D027048002","deptName":"項目管理與效能中心"}],[{"deptId":"D027","deptName":"杭州研究院"},{"deptId":"D027048","deptName":"技術工程事業(yè)部"}],[{"deptId":"D027","deptName":"杭州研究院"},{"deptId":"D027048","deptName":"技術工程事業(yè)部"}]]';
    const arr = JSON.parse(str).flat(); //扁平化

    let newArr = [];
    noRepeat(arr).length && noRepeat(arr).forEach(it => {
      appendChild(it, newArr);
    });
  }, [])

  const noRepeat = (arr) => { //去重
    let newobj = {}; 
    return arr.reduce((preVal, curVal) => {
    newobj[curVal.deptId] ? '' : newobj[curVal.deptId] = preVal.push(curVal); 
      return preVal 
    }, []);
  }

  const appendChild = (item, newArr) => {
    if(!newArr.find(it => item.deptId.indexOf(it.deptId) > -1)) { //所有一級部門
      newArr.push({
        deptId: item.deptId,
        deptName: item.deptName,
        children: [],
      });
      setTreeData(newArr);
    }else {
      appendOtherChild(item, newArr);
    }
  }

  const appendOtherChild = (item, newArr) => {
    newArr.map(it => {
      if(item.deptId.indexOf(it.deptId) > -1 && item.deptId.length === it.deptId.length+3) {
        it.children.push({
          deptId: item.deptId,
          deptName: item.deptName,
          children: [],
        })
      }else {
        appendOtherChild(item, it.children);
      }
    });
    setTreeData(newArr);
  }