js扁平数组和树结构相互转换处理方法

更新时间：2023年07月06日 11:07:37 作者：忧郁的蛋~

这篇文章主要给大家介绍了关于js扁平数组和树结构相互转换处理方法的相关资料,之前面试有遇到过这个问题,面试官问如何把一个数组数据扁平,然后转化为Tree结构数据,工作中刚好也用到了,所以总结下,需要的朋友可以参考下

一、将扁平的数据转为树形结构

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

扁平数据通常是一个带有 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) {
            // '&& pid === 0' 判断条件是为了只将 pid 为 0 作为根节点，其它值不作为根节点。
            // 如需将其它值也作为根节点，可将 'else if (!map.has(pid) && pid === 0)' 改为 'else'
            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、方法一

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

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

 
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、方法三

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

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

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）。

总结

到此这篇关于js扁平数组和树结构相互转换处理方法的文章就介绍到这了,更多相关js扁平数组和树结构相互转换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

关于javascript DOM事件模型的两件事
DOM事件模型的两件事：事件捕捉(Event Capture)的实现问题以及IE的高级事件处理模型的问题。
2010-07-07
微信小程序实现轮播图效果
这篇文章主要为大家详细介绍了微信小程序实现轮播图效果，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2017-09-09
JavaScript forEach的几种用法小结
forEach()是JavaScript中一个常用的方法,用于遍历数组或类数组对象中的每个元素,本文就来介绍一下JavaScript forEach的几种用法小结,具有一定的参考价值,感兴趣的可以了解一下
2023-11-11
js判断手机系统是android还是ios
本文主要介绍了js判断手机系统是android还是ios的方法，具有很好的参考价值。下面跟着小编一起来看下吧
2017-03-03
15个简单的JS编码标准让你的代码更整洁(小结)
这篇文章主要介绍了15个简单的JS编码标准让你的代码更整洁(小结),文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-07-07
js使用Canvas将多张图片合并成一张的实现代码
这篇文章主要介绍了js使用Canvas将多张图片合并成一张的实现代码,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-04-04
el-input 标签中密码的显示和隐藏功能的实例代码
本文通过实例代码给大家介绍了el-input 标签中密码的显示和隐藏 ,代码简单易懂，非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下
2019-07-07
如何使用ES6的class类继承来实现绚丽小球效果
JS是由ES(ECMAScript)、DOM(浏览器文档对象)、BOM(浏览器对象模型)组成,这篇文章主要给大家介绍了关于如何使用ES6的class类继承来实现绚丽小球效果的相关资料,需要的朋友可以参考下
2021-06-06
基于 antd pro 的短信验证码登录功能(流程分析)
这篇文章主要介绍了基于 antd pro 的短信验证码登录功能(流程分析),本文通过实例代码流程分析给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2021-05-05
JavaScript数组去重的几种方法效率测试
JavaScript数组去重是前端面试酷爱的问题，问题简单而又能看出程序员对计算机程序执行过程的理解如何。数组去重的方法有很多，到底哪种是最理想的我不清楚。于是我测试了下数组去重的效率。测试二十万个数据，随着数据越多效率很明显的就体验了出来。下面来一起看看吧。
2016-10-10