JS实现树形结构与数组结构相互转换并在树形结构中查找对象

Bertha ·

更新时间:2024-09-20

· 1688 次阅读

总是有很多需求是关于处理树形结构的，所以不得不总结几个常见操作的写法。

首先假设有一个树形结构数据如下

var tree=[
  {
    'id': '1',
    'name': '教学素材管理',
    'children':[
      {
        'id': '101',
        'name': '教学素材',
        'children':[
          {
            'id': '10101',
            'name': '修改',
          },
          {
            'id': '10102',
            'name': '添加',
          }
        ]
      },  {
        'id': '102',
        'name': '测试试题',
      },
      {
        'id': '103',
        'name': '问题任务',
      }
    ]
  }, {
    'id': '2',
    'name': '基础数据管理',
    'children':[
      {
        'id': '201',
        'name': '专业设置',
      },
      {
        'id': '202',
        'name': '专业管理',
      }
    ]
  }
]

1、如何在tree中找到id=10102的对象？

思路一：深度遍历，从顶点开始，当前节点有子节点则遍历当前节点的子节点（递归）。

function deepQuery(tree,id) {
    var isGet = false;
    var retNode = null;
    function deepSearch(tree,id){
        for(var i = 0; i<tree.length; i++) {
            if(tree[i].children && tree[i].children.length>0) {
                deepSearch(tree[i].children,id);
            }
            if(id === tree[i].id || isGet) {
                isGet||(retNode = tree[i]);
                isGet = true;
                break;
            }
        }
    }
    deepSearch(tree,id);
    return retNode;
}

调用：

var getNode = deepQuery(tree,10102);
console.log(getNode)

思路二：广度遍历，遍历根节点的所有子节点，再从第一个子节点开始依次遍历。

function breadthQuery(tree, id) {
    var stark = [];
    stark = stark.concat(tree);
    while(stark.length) {
        var temp = stark.shift();
        if(temp.children) {
            stark = stark.concat(temp.children);
        }
        if(temp.id === id) {
            return temp;
        }
    }
}

调用：

var getNode=breadthQuery(tree,10102);
console.log(getNode);

2、如何将树形结构转换为有父子关系属性的数组结构？

思路一：初始化一个空数组，从tree的顶端开始遍历，当前节点有子节点时，一边继续遍历子节点，一边在当前节点上删除子节点，将当前节点push到空数组。

function flatten1(tree) {
    var arr = [];
    function spread(tree,pid) {
        for (var i=0; i < tree.length; i++ ) {
            item = tree[i]
            let {id,name}=item;
            arr.push({id,name,pid})
            if (item.children) {
                spread(item.children,item.id)
                delete item.children
            }
        }
    }
    spread(tree,0)
    return arr;
}
var newArr = flatten1(tree);

思路二：

function flatten2 (data,pid) {
  return data.reduce((arr, {id, name, children = []}) =>
    arr.concat([{id, name,pid}], flatten2(children,id)), [])
} 
var newArr = flatten2(tree,0);

结果：

3、如何将数组结构转换为树形结构？

下面是偶然看到一位大佬很秀的写法（原文链接）

function treeData(data){   
    let cloneData = JSON.parse(JSON.stringify(data))
    return cloneData.filter(parent=>{
        let branchArr = cloneData.filter(child => parent['id'] == child['pid']);
        branchArr.length>0 ? parent['children'] = branchArr : '';
        return parent['pid'] == 0 ;
    })
}

调用：

var newTree = treeData(newArr)

用之前测试生成的数组试一下