树形数据扁平化可通过递归或迭代实现,递归适合一般场景,迭代避免栈溢出,添加level字段可保留层级信息,便于前端展示。
树形数据扁平化是前端开发中常见的需求,比如将嵌套的菜单、组织架构或分类数据转换成一维数组,方便渲染或搜索。JavaScript 提供了多种方式实现这一操作,核心思路是遍历树结构,把每个节点按顺序提取出来。
递归是最直观的方法,通过深度优先遍历,依次访问每个子节点,并将其推入结果数组。
示例代码:
function flattenTree(tree, result = []) {
tree.forEach(node => {
result.push({ ...node });
if (node.children) {
flattenTree(node.children, result);
}
});
return result;
}
说明:
对于层级非常深的树,递归可能导致调用栈溢出。可以用栈结构手动维护遍历过程。
function flattenTreeIteratively(tree) {
const result = [];
const stack = [...tree];
while (stack.length > 0) {
const node = stack.pop();
result.unshift(node); // 保证顺序可改为 unshift 或 reverse
if (node.children) {
stack.push(...node.children);
}
}
return result;
}
注意:
在实际应用中,可能需要保留节点的层级关系,比如用于缩进显示。
function flattenWithLevel(tree, level = 0) {
let result = [];
tree.forEach(node => {
result.push({ ...node, level });
if (node.children) {
result = result.concat(flattenWithLevel(node.children, level + 1));
}
});
return result;
}
这样每个节点都会带上 level 字段,前端可以根据 level 添加缩进样式。
基本上就这些。根据具体需求选择递归或迭代方式,是否需要保留层
级信息也很关键。扁平化本身不复杂,但处理好边界和性能能让代码更健壮。