JS中树的查找：通过子节点ID查找所有关联的父节点

2023-02-28

函数节点 id

提示：底部有完整的源码，童鞋们有需要可以CTRL+C拿走，但我建议你能真正理解实现原理，而不是直接CTRL+C。树是JS中非常常见的数据结构。知识库目录，侧边栏菜单，字典目录，企业组织架构等都可能会用到树的操作。不知道大家有没有使用过element-ui等组件库中的级联选择器?它所接收的数据类型就是

提示：底部有完整的源码，童鞋们有需要可以CTRL + C拿走，但我建议你能真正理解实现原理，而不是直接CTRL + C。

树是JS中非常常见的数据结构。知识库目录，侧边栏菜单，字典目录，企业组织架构等都可能会用到树的操作。

不知道大家有没有使用过element-ui等组件库中的级联选择器?它所接收的数据类型就是树形结构的数组。级联选择器的绑定值是被选中的一系列父子节点id构成的数组，而后端通常仅需要我们提供最后一个叶子节点的id;然而，在编辑的时候，我们仅提供最后一个叶子节点的id是无法还原级联选择器选项的选中状态的;所以，我们需要通过该节点id将与其关联的父节点全部查找出来，得到一个与级联选择器适配的数据。

现在，我们实现根据节点id查找关联的父节点函数getTreeIds。我们先定义一个空函数，该函数需要接收3个参数：树形结构数组tree，节点id，节点信息配置config。config用于配置id和children字段，因为这是一个通用函数，我们需要应对后端开发者使用的children和id字段可能存在的不统一问题。

export const getTreeIds = (tree, nodeId, config) => {}1.

下面的所有代码全部写在getTreeIds函数体内。现在，我们从config中拿到children和id字段，并分别设置一个默认值。

const { children = 'children', id = 'id' } = config || {}1.

假设我们的数据结构是下面这样的，该如何通过子节点查找父节点呢?我们都知道，父子节点的关联是通过children字段建立的。可惜这个关联是单向的，我们只能通过父节点查找子节点，而不能通过子节点查找父节点，因为我们的子节点缺少对父节点的引用。

[
  {
    id: 1,
    label: 'test1',
    children: [
      {
        id: 2,
        label: 'test1-1',
        children: [ { id: 3,  label: 'test1-1-1'  }]
      }
    ]
  }
]1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.

我们需要编写一个函数手动将子节点与父节点建立绑定。最简单的查找方式就是将树形结构转化为扁平化数组，并建立关联，然后再进行查找。

toFlatArray函数的源码如下，我们使用数组的reduce方法对树形结构数组进行聚合递归转化为扁平化的树，将parentId字段添加到子节点，这样我们就建立了子节点到父节点的关联。

const toFlatArray = (tree, parentId) => {
    return tree.reduce((t, _) => {
      const child = _[children]
      return [
        ...t,
        parentId ? { ..._, parentId } : _,
        ...(child && child.length ? toFlatArray(child, _[id]) : [])]
    }, [])
  }1.
2.
3.
4.
5.
6.
7.
8.
9.

然后，我们创建getIds函数，该函数接收一个扁平化的树，使用while循环进行查找，如果存在parentId，我们就把它添加到ids数组的开头，一直查找到树的根部。最后，我们返回ids数组。

const getIds = flatArray => {
    let ids = [nodeId]
    let child = flatArray.find(_ => _[id] === nodeId)
    while (child && child.parentId) {
      ids = [child.parentId, ...ids]
      child = flatArray.find(_ => _[id] === child.parentId)
    }
    return ids
  }1.
2.
3.
4.
5.
6.
7.
8.
9.

最后，我们在getTreeIds函数体的尾部，将标准树转化为扁平化的树后传递给getIds函数返回的ids数组返回。

return getIds(toFlatArray(tree))1.

现在，我们大功告成了。我们可以用一些假数据测试一下：

const treeData = [
  {
    id: 1,
    label: 'test1',
    children: [
      {
        id: 2,
        label: 'test1-1',
        children: [
          {
            id: 3,
            label: 'test1-1-1'
          },
          {
            id: 4,
            label: 'test1-1-2',
            children: [
              {
                id: 5,
                label: 'test1-1-1-1'
              }
            ]
          }
        ]
      }
    ]
  }
]
console.log(getTreeIds(treeData, 5)) // 输出 [1, 2, 4, 5]
console.log(getTreeIds(treeData, 3)) // 输出 [1, 2, 3]1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.
23.
24.
25.
26.
27.
28.
29.
30.