您现在的位置是：亿华云 > IT科技
JS中树的查找：通过子节点ID查找所有关联的父节点
亿华云2025-10-03 02:22:25【IT科技】1人已围观
简介提示：底部有完整的源码，童鞋们有需要可以CTRL + C拿走，但我建议你能真正理解实现原理，而不是直接CTRL + C。树是JS中非常常见的数据结构。知识库目录，侧边栏菜单，字典目录，企业组织架构等都
提示：底部有完整的中树找通找源码，童鞋们有需要可以CTRL + C拿走，查点I点但我建议你能真正理解实现原理，过节而不是有关直接CTRL + C。
树是父节JS中非常常见的数据结构。知识库目录，中树找通找侧边栏菜单，查点I点字典目录，过节企业组织架构等都可能会用到树的有关操作。
不知道大家有没有使用过element-ui等组件库中的父节级联选择器?它所接收的数据类型就是树形结构的数组。级联选择器的中树找通找绑定值是被选中的一系列父子节点id构成的数组，而后端通常仅需要我们提供最后一个叶子节点的查点I点id;然而，在编辑的过节时候，我们仅提供最后一个叶子节点的有关id是无法还原级联选择器选项的选中状态的;所以，我们需要通过该节点id将与其关联的亿华云计算父节父节点全部查找出来，得到一个与级联选择器适配的数据。
现在，我们实现根据节点id查找关联的父节点函数getTreeIds。我们先定义一个空函数，该函数需要接收3个参数：树形结构数组tree，节点id，节点信息配置config。config用于配置id和children字段，因为这是一个通用函数，我们需要应对后端开发者使用的children和id字段可能存在的不统一问题。
export const getTreeIds = (tree, nodeId, config) => { }下面的所有代码全部写在getTreeIds函数体内。现在，我们从config中拿到children和id字段，并分别设置一个默认值。
const { children = children, id = id } = config || { }假设我们的数据结构是下面这样的，该如何通过子节点查找父节点呢?我们都知道，父子节点的关联是通过children字段建立的。可惜这个关联是单向的，服务器托管我们只能通过父节点查找子节点，而不能通过子节点查找父节点，因为我们的子节点缺少对父节点的引用。
[
{
id: 1,
label: test1,
children: [
{
id: 2,
label: test1-1,
children: [ { id: 3, label: test1-1-1 }]
}
]
}
]我们需要编写一个函数手动将子节点与父节点建立绑定。最简单的查找方式就是将树形结构转化为扁平化数组，并建立关联，然后再进行查找。
toFlatArray函数的源码如下，我们使用数组的reduce方法对树形结构数组进行聚合递归转化为扁平化的树，将parentId字段添加到子节点，这样我们就建立了子节点到父节点的关联。
const toFlatArray = (tree, parentId) => {
return tree.reduce((t, _) => {
const child = _[children]
return [
...t,
parentId ? { ..._, parentId } : _,
...(child && child.length ? toFlatArray(child, _[id]) : [])]
}, [])
}然后，我们创建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
}最后，我们在getTreeIds函数体的网站模板尾部，将标准树转化为扁平化的树后传递给getIds函数返回的ids数组返回。
return getIds(toFlatArray(tree))现在，我们大功告成了。我们可以用一些假数据测试一下：
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]童鞋们，学会了吗?是不是很简单?
很赞哦!（55）
