数组与树形结构转换


        const arr3 = [
            { id: 1, name: "部门1", pid: 0 },
            { id: 2, name: "部门2", pid: 1 },
            { id: 3, name: "部门3", pid: 1 },
            { id: 4, name: "部门4", pid: 3 },
            { id: 5, name: "部门5", pid: 4 },
            { id: 6, name: "部门6", pid: 5 },
            { id: 7, name: "部门7", pid: 6 },
        ]
        /**
         * 扁平数据结构转树结构数据
         * @param arr 扁平数据
         * @param topPid 非必填 顶部pid
         * @returns 树结构数据
         */
        const arr2Tree = (arr3, topPid = undefined) => {
            const arrMap = new Map()
            const newArr = []
            let newTopPid = topPid
            if (newTopPid === undefined) {
                newTopPid = arr3.sort((a, b) => a.pid - b.pid)[0].pid
            }

            arr3.forEach((item, index) => {
                if (item.pid === newTopPid) {
                    newArr.push(item)
                }
                if (!arrMap[item.id]) {
                    arrMap[item.id] = item
                }
                if (arrMap[item.pid]) {
                    if (!arrMap[item.pid].children) {
                        arrMap[item.pid].children = []
                    }
                    arrMap[item.pid].children.push(item)
                }
            })
            return newArr
        }
        const tree = arr2Tree(arr3)
        console.log("tree", tree)
        /**
         * 树结构数据转扁平结构数据
         * @param tree 树结构数据
         * @returns 扁平结构数据
         */
        const tree2Arr = (tree) => {
            const newArr = []
            /**
             * 递归
             */
            const recursive = (arr) => {
                arr.forEach((item) => {
                    newArr.push(item)
                    if (item.children?.length) {
                        recursive(item.children)
                    }
                })
            }
            recursive(tree)
            return newArr
        }
        const newArr = tree2Arr(tree)
        console.log("newArr", newArr)

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值