博客

先新建一个数组

 const arr = [{
            pid: null,
            id: 1,
            name: '北京'
        },
        {
            pid: 1,
            id: 11,
            name: '朝阳'
        },
        {
            pid: 11,
            id: 111,
            name: '朝阳1号'
        },
        {
            pid: 1,
            id: 12,
            name: '海淀'
        },
        {
            pid: 12,
            id: 121,
            name: '海淀1号'
        },
        {
            pid: null,
            id: 2,
            name: '上海'
        },
        {
            pid: 2,
            id: 21,
            name: '浦东'
        },
        {
            pid: 21,
            id: 211,
            name: '浦东1号'
        },
        {
            pid: 2,
            id: 22,
            name: '虹口'
        },
        {
            pid: 22,
            id: 221,
            name: '虹口1号'
        }
    ]

树形结构转平铺结构

function treeToArr(arr, pid = null, res = []) {
        arr.forEach(item => {
            res.push({
                pid: pid,
                id: item.id,
                name: item.name
            });
            if (item.children && item.children.length !== 0) {
                treeToArr(item.children, item.id, res);
            }
        });
        return res;
    };

    const aaaa = treeToArr(arr);
    console.log(aaaa);

平铺结构转树形结构

function arrToTree(arr, pid = null) {
        const res = [];
        arr.forEach(item => {
            if (item.pid === pid) {
                // 这样每次都需要遍历整个数组，因此时间复杂度为 n*n
                // const children = arrToTree(arr, item.id)   

                // 往下递归时，每次数组的大小都在减小，每次都筛选掉父代元素，最终的时间复杂度为 n*logn
                const children = arrToTree(arr.filter(v => v.pid !== pid), item.id);
                if (children.length) {
                    res.push({
                        ...item,
                        children
                    })
                } else {
                    res.push({
                        ...item
                    })
                }
            }
        });
        return res;
    };

    const tree = arrToTree(arr);
    console.log(JSON.stringify(tree, null, 2));