以树形结构实现省市区扁平化
在数据结构中，省市区通常以树形结构呈现，但有时需要将其扁平化为一级二级三级代码的结构。
问题描述
给定省市区树形数据，其中节点带有已选中的标记：“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：检查某个市的县级子节点选中情况，并返回一个扁平化的数组，其中每个元素包含省级、市级和县级代码。
代码说明
- getNewData 函数遍历一级区域，如果一级区域被选中，则初始化一个对象 obj 表示一级区域信息。
- cityCheck 函数遍历二级区域，如果二级区域被选中，则更新 obj 对象的二级区域信息。
- countryCheck 函数遍历三级区域，如果三级区域被选中，则将三级区域信息添加到 obj 对象中。
- 根据二级区域和三级区域的选中情况，更新 obj 对象的 cityAreald、countryAreald 和 actualAreaLevel 属性。
- 将处理后的 obj 对象添加到结果数组 d 中。
