如何将嵌套的省市区树结构扁平化为指定格式，以满足不同选择的地址获取需求？

省市区树结构扁平化转换结构

本文旨在解决如何将嵌套的省市区结构树扁平化为指定格式的结构，以满足特定需求。

需求：

• 如果三级地址都选中了，只获取一级和二级地址。
• 如果二级地址都选中了，只获取一级地址。
• 如果三级地址选中了某些，获取一级、二级和三级地址。

数据结构：

树形结构的数据示例如下：

ChatX翻译

最实用、可靠的社交类实时翻译工具。 支持全球主流的20+款社交软件的聊天应用，全球200+语言随意切换。 让您彻底告别复制粘贴的翻译模式，与世界各地高效连接！

下载

[
  {
    "code": "110000",
    "value": "北京市",
    "checked": "1",
    "children": [
      {
        "code": "110100",
        "value": "北京市",
        "checked": "1",
        "children": [
          {
            "code": "110101",
            "value": "东城区",
            "checked": "1"
          },
          {
            "code": "110102",
            "value": "西城区",
            "checked": "1"
          }
        ]
      }
    ]
  },
  {
    "code": "120000",
    "value": "天津市",
    "checked": "1",
    "children": [
      {
        "code": "120100",
        "value": "天津市",
        "checked": "1",
        "children": [
          {
            "code": "120101",
            "value": "和平区",
            "checked": "0"
          },
          {
            "code": "120102",
            "value": "河东区",
            "checked": "1"
          }
        ]
      }
    ]
  }
]

扁平化处理：

以下是扁平化处理的 JavaScript 代码：

function getNewData(data) {
  let d = []

  for (let province of data) {

    if (province.checked == 1) {
      let obj = {
        provinceAreald: province.code,
        cityAreald: null,  //如2级全部选中为null
        countryAreald: null,  //如3级全部选中为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级菜单被<部分>选中
      }

    }
  }


  function cityCheck(province, obj, d) {

    let cityArr = []
    for (let city of province.children) {

      if (city.checked == 1) {
        Object.assign(obj, {
          cityAreald: city.code,
          actualAreaLevel: '2',
        })

        // 参数obj, d可能被改变
        const countryArr = countryCheck(city, obj, d)

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

      }

    }
    return cityArr

  }


  function countryCheck(city, obj, d) {

    let countryArr = []
    for (let country of city.children) {

      if (country.checked == 1) {
        countryArr.push(
          Object.assign(obj, {
            countryAreald: country.code,
            actualAreaLevel: '3',
          })
        )
      }

    }
    return countryArr
  }


  return d
}

const newData = getNewData(data)
console.log(newData)

扁平化结果：

[
  {
    provinceAreald : '110000',
    cityAreald: null,  //如2级全部选中为null
    countryAreald: null,  //如3级全部选中为null
    actualAreaLevel:'1'
  },
  {
    provinceAreald : '120000',
    cityAreald: '120100',  //如2级全部选中为null
    countryAreald: '120102',  //如3级全部选中为null
    actualAreaLevel:'3' //层级
  }
]