最新要闻

广告

手机

iphone11大小尺寸是多少?苹果iPhone11和iPhone13的区别是什么?

iphone11大小尺寸是多少?苹果iPhone11和iPhone13的区别是什么?

警方通报辅警执法直播中被撞飞:犯罪嫌疑人已投案

警方通报辅警执法直播中被撞飞:犯罪嫌疑人已投案

家电

全球报道:JavaScript 扁平与树形数组数据的转换

来源:博客园
一、扁平转树形

在 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)。

关键词: 树形结构 递归算法 可以使用