JavaScript 扁平与树形数组数据的转换

更新日期: 2023-01-12阅读: 1.1k标签: 

一、扁平转树形

在 JavaScript 中,可以使用递归算法将扁平的数据转换为树形结构。

扁平数据通常是一个带有 parentId 属性的数组,而树形结构通常是一个带有 children 属性的对象。

1、方法一

下面是一个简单的例子,它演示了如何将扁平数组转换为树形对象:

let data = [
    { id: 1, pid: 0, name: '沃尔玛' },
    { id: 2, pid: 0, name: '生鲜区' },
    { id: 3, pid: 1, name: '日用品区' },
    { id: 4, pid: 2, name: '鱼' },
    { id: 5, pid: 2, name: '牛肉' },
    { id: 6, pid: 13, name: '卫生纸' },
    { id: 7, pid: 3, name: '牙刷' },
    { id: 8, pid: 7, name: '电动牙刷' },
    { id: 9, pid: 7, 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) {
            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 },
    { id: 2, name: 'Node 2', parentId: 1 },
    { id: 3, name: 'Node 3', parentId: 2 },
    { id: 4, name: 'Node 4', parentId: 3 },
    { id: 5, 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, name: 'Node 1', parentId: null },
    { id: 2, name: 'Node 2', parentId: null },
    { id: 3, name: 'Node 3', parentId: 1 },
    { id: 4, name: 'Node 4', parentId: 2 },
    { id: 5, name: 'Node 5', parentId: 2 }
];

const tree = _(flatData)
    .groupBy('parentId')
    .mapValues((children, parentId) => ({
        id: parentId || 'root',
        children: children.map(({ id, name, parentId }) => ({ id, name, parentId }))
    }))
    .values()
    .value();

console.log(tree)

在这种情况下,假设parentId为null的数据项是根节点,那么所有其它的数据项的 parentId 分别对应它的父节点,我们可以使用 groupBy() 来将所有节点根据它们的 parentId 分组,然后我们可以使用 mapValues() 来构造每个组的新结构。

4、方法四


const flatData = [
    { id: 1, name: 'Node 1', value: 1, parentId: null },
    { id: 2, name: 'Node 2', value: 2, parentId: 1 },
    { id: 3, name: 'Node 3', value: 3, parentId: 2 },
    { id: 4, name: 'Node 4', value: 4, parentId: 3 },
    { id: 5, name: 'Node 5', 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, { label: item[nameName], id: item[valueName] }) : item);
        }
    });
    return treeData;
}

const treeData = convertToTree(flatData, 'id', 'parentId', 'name', 'value');
console.log(treeData);


、树形转扁平

1、方法一

在 JavaScript 中,可以使用递归算法将树形结构数据转换为扁平数组。

下面是一个简单的例子,它演示了如何将树形数据转换为扁平数组:


const treeData = [{
    id: 1,
    name: 'Node 1',
    children: [
        { id: 2, name: 'Node 2', children: [{ id: 3, name: 'Node 3' }, { id: 4, name: 'Node 4' }] },
        { id: 5, 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,
        name: 'Node 1',
        children: [
            {
                id: 2,
                name: 'Node 2',
                children: [
                    { id: 3, name: 'Node 3' },
                    { id: 4, name: 'Node 4' },
                ]
            },
            { id: 5, 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、方法三

在 JavaScript 中,可以使用递归算法将树形结构数据转换为扁平数组。

下面是一个简单的例子,它演示了如何将树形数据转换为扁平数组:

let data = [{
    id: 1, pid: 0, name: '沃尔玛', childrens: [
        {
            id: 2, pid: 1, name: '生鲜区', childrens: [
                { id: 4, pid: 2, name: '鱼' },
                { id: 5, pid: 2, name: '牛肉' }
            ]
        },
        {
            id: 3, pid: 1, name: '日用品区', childrens: [
                { id: 6, pid: 3, name: '卫生纸' },
                { id: 7, pid: 3, 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)。

链接: https://fly63.com/article/detial/12314

JavaScript实现二叉排序树

Js二叉树排序实现:1初始化二叉树,2二叉树的遍历,3查找最小值,4查找最大值,5删除节点

js “指针”:数组转树

当变量指向一个对象的时候,实际指向的是存储地址,数组转树的方式:第一次遍历将数组转节点对象,存储到新的对象里,id为键值方便索引,第二次遍历根据索引插入子节点

Js实现二叉搜索树

计算机科学中最常用和讨论最多的数据结构之一是二叉搜索树。这通常是引入的第一个具有非线性插入算法的数据结构。二叉搜索树类似于双链表,每个节点包含一些数据,以及两个指向其他节点的指针;它们在这些节点彼此相关联的方式上有所不同

js 实现 list转换成tree(数组到树)

JS 将有父子关系的平行数组转换成树形数据:方法一:双重遍历,一次遍历parentId,一次遍历id == parendId;该方法应该能很容易被想到,实现起来也一步一步可以摸索出来;

Js算法之自平衡树

节点的高度和平衡因子;节点高度:从节点到任意子节点的彼岸的最大值。这个相对来说容易理解。那么获得节点高度的代码实现如下:平衡因子:每个节点左子树高度和右子树高度的差值。该值为0 、 -1、 1 时则为正常值

vue递归组件:树形控件

在编写树形组件时遇到的问题:组件如何才能递归调用?递归组件点击事件如何传递?组件目录及数据结构;在组件模板内调用自身必须明确定义组件的name属性,并且递归调用时组件名称就是name属性

js中的AST

在计算机科学中,抽象语法树(Abstract Syntax Tree,AST),或简称语法树(Syntax tree),是源代码语法结构的一种抽象表示。它以树状的形式表现编程语言的语法结构,树上的每个节点都表示源代码中的一种结构

Js二叉树的遍历

二叉树是每个结点最多有两个子树的树结构。通常子树被称作“左子树”(left subtree)和“右子树”(right subtree)。二叉树常被用于实现二叉查找树和二叉堆。

js将扁平结构数据转换为树形结构

最近项目又频繁需要对扁平结构进行树形转换,这个算法从我最早接触的时候使用了递归,到现在的单次循环完成,简单记录一下算法的演变,算是对树形算法的一个简单记录,这种类型的算法在项目中的使用挺多的

快速实现一个简单可复用可扩展的Vue树组件

大概因为平时工作项目的原因,写了很多次树形组件,越写越觉得可以写得更简单并且更具有复用性、扩展性。树组件的应用场景很多,比如一篇文章的目录、一个公司部门组织情况、思维导图等,其实都可以用树形结构来描述

点击更多...

内容以共享、参考、研究为目的,不存在任何商业目的。其版权属原作者所有,如有侵权或违规,请与小编联系!情况属实本人将予以删除!