一、将扁平的数据转为树形结构
在 js中,可以使用递归算法将扁平的数据转换为树形结构。
扁平数据通常是一个带有 parentId 属性的数组,而树形结构通常是一个带有 children 属性的对象。
1、方法一
下面是一个简单的例子,演示如何将扁平数组转换为树形对象:
let data = [
{ id: 1,pid: 0,name: '中国' },{ id: 2,name: '巴铁' },{ id: 3,pid: 1,name: '小日本儿' },{ id: 4,pid: 2,name: '杂种美国' },{ id: 5,name: '俄罗斯' },{ id: 6,pid: 13,name: '乌克兰' },{ id: 7,pid: 3,name: '英国' },{ id: 8,pid: 7,name: '意大利' },{ id: 9,name: '西班牙' }
];
function convertToTree(flatData) {
let treeData = [];
let map = new Map();
let outputObj,pid;
for (let i = 0; i < flatData.length; i++) {
pid = flatData[i].pid;
if (map.has(pid)) {
if (!map.get(pid).childrens)
map.get(pid).childrens = [];
let obj = new Object(flatData[i]);
map.get(pid).childrens.push(obj);
map.set(flatData[i].id,obj);
} else if (!map.has(pid) && pid === 0) {
// '&& pid === 0' 判断条件是为了只将 pid 为 0 作为根节点,其它值不作为根节点。
// 如需将其它值也作为根节点,可将 'else if (!map.has(pid) && pid === 0)' 改为 'else'
outputObj = new Object(flatData[i]);
treeData.push(outputObj);
map.set(flatData[i].id,outputObj);
}
}
return treeData;
}
let TreeData = convertToTree(data);
console.log(TreeData);
2、方法二
使用递归算法将扁平数组转换为树形对象:
const flatData = [
{ id: 1,name: 'Node 1',parentId: null },name: 'Node 2',parentId: 1 },name: 'Node 3',parentId: 2 },name: 'Node 4',parentId: 3 },name: 'Node 5',parentId: 3 }
];
function convertToTree(flatData,parentId = null) {
const children = flatData.filter(node => node.parentId === parentId);
if (!children.length) {
return null;
}
return children.map(node => ({
...node,children: convertToTree(flatData,node.id)
}));
}
const treeData = convertToTree(flatData);
console.log(treeData);
该算法的流程为:
- 使用 filter() 函数过滤出所有的子节点。
- 使用 map() 函数构造每个子节点的新结构,并使用递归来处理子节点的 children 属性。
3、方法三
还有一些第三方库可以帮助你转换扁平数据为树形结构,例如 lodash
中的 _.groupBy()
和 _.mapValues()
方法可以帮助你将扁平数据转换为树形数据。
const flatData = [
{ id: 1,parentId: 2 }
];
const tree = _(flatData)
.groupBy('parentId')
.mapValues((children,parentId) => ({
id: parentId || 'root',children: children.map(({ id,name,parentId }) => ({ id,parentId }))
}))
.values()
.value();
console.log(tree)
在这种情况下,假设parentId为null的数据项是根节点,那么所有其它的数据项的 parentId 分别对应它的父节点,我们可以使用 groupBy()
来将所有节点根据它们的 parentId 分组,然后我们可以使用 mapValues()
来构造每个组的新结构。
4、方法四
const flatData = [
{ id: 1,value: 1,value: 2,value: 3,value: 4,value: 5,parentId: 3 }
];
/**
* 将扁平数组转换为树形对象,
* 适用于来自同一个表的数据,即idName的数据不重复
*
* @param {*} flatData 同级数组数据
* @param {*} idName 唯一id
* @param {*} pidName 父级id
* @param {*} nameName 自定义(Cascader 级联选择器)键名称
* @param {*} valueName 自定义(Cascader 级联选择器)值名称
* @returns 树形结构数组数据,适用于Cascader 级联选择器组件
*/
function convertToTree(flatData,idName,pidName,nameName,valueName) {
let treeData = [];
if (!Array.isArray(flatData)) {
return treeData;
}
flatData.forEach((item) => {
delete item.children;
});
let map = {};
flatData.forEach((item) => {
// 深拷贝,该方式将使该转换方法失效
// map[item[idName]] = JSON.parse(JSON.stringify(item));
// 浅拷贝,将对 item 的引用传递给 map
map[item[idName]] = item;
});
flatData.forEach((item) => {
// 深拷贝,该方式将使该转换方法失效
// let parent = JSON.parse(JSON.stringify(map[item[pidName]]));
// 浅拷贝,将对 map 的引用传递给 parent
let parent = map[item[pidName]];
if (parent) {
// 利用了浅拷贝的引用传递,最终 flatData 中的 item 将会改变
(parent.children || (parent.children = [])).push(
nameName && valueName ? Object.assign(item,{ label: item[nameName],id: item[valueName] }) : item);
} else {
treeData.push(
nameName && valueName ? Object.assign(item,id: item[valueName] }) : item);
}
});
return treeData;
}
const treeData = convertToTree(flatData,'id','parentId','name','value');
console.log(treeData);
二、树形转扁平
1、方法一
在 js中,可以使用递归算法将树形结构数据转换为扁平数组。
下面是一个简单的例子,它演示了如何将树形数据转换为扁平数组:
const treeData = [{
id: 1,children: [
{ id: 2,children: [{ id: 3,name: 'Node 3' },name: 'Node 4' }] },name: 'Node 5' }
]
}];
function convertToFlat(data,parentId = null) {
return data.reduce((acc,curr) => {
acc.push({ ...curr,parentId });
if (curr.children) {
acc = acc.concat(convertToFlat(curr.children,curr.id));
}
return acc;
},[]);
}
const flatData = convertToFlat(treeData);
console.log(flatData);
该算法的流程为:
- 使用 reduce() 函数遍历每个节点,并将父节点的 id 作为参数传递给递归函数。
- 使用 push() 函数将当前节点添加到结果数组中。
- 使用 concat() 函数将递归调用的结果与结果数组连接在一起。
- 如果当前节点有 children 属性,则递归调用 convertToFlat() 函数,并将当前节点的 id 作为父节点传递给函数。
注意:该方法返回的扁平结构数据未将 children属性删除,因此存在冗余的数据。
这是一种将树形结构数据转换为扁平数组的方法,如果有其他特定的需求,还可以使用其他方法来转换数据,例如使用广度优先遍历算法,使用队列存储节点。
2、方法二
const treeData = [
{
id: 1,children: [
{
id: 2,children: [
{ id: 3,name: 'Node 4' },]
},name: 'Node 5' },]
},{
id: 6,name: 'Node 6',children: [
{ id: 7,name: 'Node 7' }
]
},];
function convertToFlat(treeData,parentId = null) {
let flatData = [];
for (let node of treeData) {
flatData.push({ id: node.id,name: node.name,parentId });
if (node.children) {
flatData = flatData.concat(convertToFlat(node.children,node.id));
}
}
return flatData;
}
const flatData = convertToFlat(treeData);
console.log(flatData);
该算法的流程为:
- 创建一个空的扁平数组。
- 递归遍历树形数组中的每个节点,将当前节点添加到扁平数组中。
- 对于当前节点的子节点,继续使用递归,并将子节点添加到扁平数组中。
- 返回扁平数组
注意:该方法需要手动构造push 到扁平数组的对象,通用性较差。
3、方法三
在 js中,可以使用递归算法将树形结构数据转换为扁平数组。
下面是一个简单的例子,它演示了如何将树形数据转换为扁平数组:
let data = [{
id: 1,name: '棒子',childrens: [
{
id: 2,name: '中国',childrens: [
{ id: 4,name: '杂种美国' }
]
},{
id: 3,name: '俄罗斯',childrens: [
{ id: 6,name: '阿三' }
]
}
]
}];
function convertToFlat(treeData) {
let flatData = [];
for (let i = 0; i < treeData.length; i++) {
if (treeData[i].childrens) {
flatData.push(...convertToFlat(treeData[i].childrens));
delete treeData[i].childrens;
}
flatData.push({ ...treeData[i] });
}
return flatData;
}
let flatData = convertToFlat(data);
console.log(flatData);
该方法通用性较强,对树形结构数组数据内部具体的属性名称要求较小(除 childrens)。