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

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

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

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

问题描述

给定省市区树形数据,其中节点带有已选中的标记:“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 中。

以上就是如何将省市区树形结构扁平化并根据区域选中情况获取对应代码?的详细内容,更多请关注www.sxiaw.com其它相关文章!