如何将省市区树形结构扁平化并根据区域选中情况获取对应代码？

以树形结构实现省市区扁平化

在数据结构中，省市区通常以树形结构呈现，但有时需要将其扁平化为一级二级三级代码的结构。

问题描述

给定省市区树形数据，其中节点带有已选中的标记：“checked”。要求将树形结构扁平化为：

• 如果三级区域均已选中，则仅获取一级和二级区域代码，三级区域代码为 null。
• 如果二级区域均已选中，则仅获取一级区域代码，二级和三级区域代码为 null。
• 如果三级选中部分区域，则获取一级，二级和三级区域代码。

解决方案

以下 JavaScript 代码实现了这一转换：

function getNewData(data) {
  let d = []

  for (let province of data) {
    if (province.checked == 1) {
      let obj = {
        provinceAreald: province.code,
        cityAreald: null,
        countryAreald: null,
        actualAreaLevel: '1'
      }

      const cityArr = cityCheck(province, obj, d)

      if (cityArr.length == province.children.length) {
        Object.assign(obj, {
          cityAreald: null,
          actualAreaLevel: '1'
        })
        d.push(obj)  // 2级菜单被<全部>选中
      } else {
        d.push(...cityArr)  // 2级菜单被<部分>选中
      }

    }
  }
}

辅助函数说明

• cityCheck：检查某个省的市级子节点选中情况，并返回一个扁平化的数组，其中每个元素包含省级、市级和县级代码。
• countryCheck：检查某个市的县级子节点选中情况，并返回一个扁平化的数组，其中每个元素包含省级、市级和县级代码。

代码说明

1. getNewData 函数遍历一级区域，如果一级区域被选中，则初始化一个对象 obj 表示一级区域信息。
2. cityCheck 函数遍历二级区域，如果二级区域被选中，则更新 obj 对象的二级区域信息。
3. countryCheck 函数遍历三级区域，如果三级区域被选中，则将三级区域信息添加到 obj 对象中。
4. 根据二级区域和三级区域的选中情况，更新 obj 对象的 cityAreald、countryAreald 和 actualAreaLevel 属性。
5. 将处理后的 obj 对象添加到结果数组 d 中。