用reduce 方法将树结构数组转为扁平数组

interface TreeNode {
  id: string;
  children?: TreeNode[];
}

const flattenTree = (tree: TreeNode[]): TreeNode[] => {
  return tree.reduce((result: TreeNode[], node: TreeNode) => {
    result.push(node);
    if (node.children) {
      result.push(...flattenTree(node.children));
    }
    return result;
  }, []);
};

函数接收一个树结构数组tree作为参数,返回一个扁平化后的数组。
在函数内,使用Array.reduce方法和递归调用来将每个节点和其子节点转换为扁平数组中的一个元素。

Array.reduce方法的回调函数中,首先将当前节点node推入result数组。然后如果节点有子节点,就使用...展开运算符将子节点的扁平数组插入result数组中。
最后,返回result数组。

例如:

const tree: TreeNode[] = [
  {
    id: '1',
    children: [
      { id: '2' },
      { id: '3', children: [{ id: '4' }, { id: '5' }] },
      { id: '6' },
    ],
  },
  { id: '7' },
];

const flat = flattenTree(tree);
console.log(flat);
// Output: [{ id: '1' }, { id: '2' }, { id: '3' }, { id: '4' }, { id: '5' }, { id: '6' }, { id: '7' }]

树结构数组包含7个节点,其中3个节点有子节点。转换为扁平数组后,其中包含了所有的节点。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值