JavaScript實(shí)現(xiàn)無限級(jí)遞歸樹的示例代碼

const data = [
  {
    "area_id": 5,
    "name": "廣東省",
    "parent_id": 0,
  }, 
  {
    "area_id": 6,
    "name": "廣州市",
    "parent_id": 5,
  },
  {
    "area_id": 7,
    "name": "深圳市",
    "parent_id": 5,
  },
  {
    "area_id": 4,
    "name": "北京市",
    "parent_id": 3,
  },
  {
    "area_id": 3,
    "name": "北京",
    "parent_id": 0,
  },
  {
    "area_id": 2,
    "name": "測(cè)試子地區(qū)",
    "parent_id": 1,
  },
  {
    "area_id": 1,
    "name": "測(cè)試地區(qū)",
    "parent_id": 0,
  }
]

function toTreeData(data,pid){
 
  function tree(id) {
    let arr = []
    data.filter(item => {
      return item.parent_id === id;
    }).forEach(item => {
      arr.push({
        area_id: item.area_id,
        label: item.name,
        children: tree(item.area_id)
      })
    })
    return arr
  }
  return tree(pid) // 第一級(jí)節(jié)點(diǎn)的父id，是null或者0，視情況傳入
}

function setTreeData(arr) {
  // 刪除所有 children,以防止多次調(diào)用
  arr.forEach(function (item) {
      delete item.children;
  });
  let map = {}; // 構(gòu)建map
  arr.forEach(i => {
    map[i.area_id] = i; // 構(gòu)建以area_id為鍵 當(dāng)前數(shù)據(jù)為值
  });

  let treeData = [];
  arr.forEach(child => {
    const mapItem = map[child.parent_id]; // 判斷當(dāng)前數(shù)據(jù)的parent_id是否存在map中

    if (mapItem) { // 存在則表示當(dāng)前數(shù)據(jù)不是最頂層數(shù)據(jù)
    
      // 注意: 這里的map中的數(shù)據(jù)是引用了arr的它的指向還是arr，當(dāng)mapItem改變時(shí)arr也會(huì)改變,踩坑點(diǎn)
      (mapItem.children || ( mapItem.children = [] )).push(child); // 這里判斷mapItem中是否存在children, 存在則插入當(dāng)前數(shù)據(jù), 不存在則賦值children為[]然后再插入當(dāng)前數(shù)據(jù)
    } else { // 不存在則是組頂層數(shù)據(jù)
      treeData.push(child);
    }
  });

  return treeData;
};

console.log(setTreeData(data)); // 輸出整理后的數(shù)據(jù)