本文详细介绍了在JavaScript中计算非二叉树节点深度（或层级）的两种递归方法。通过构建具有名称和子节点数组的通用树结构，教程演示了如何从根节点向下搜索目标节点，以及如何从目标节点向上追溯至根节点来确定其深度。文章提供了清晰的代码示例、详细的递归逻辑解析及使用注意事项，旨在帮助开发者高效地处理树形数据。
树节点深度概念
在树形数据结构中，节点的“深度”（depth）或“层级”（level）是一个核心概念。通常，根节点的深度定义为0，其直接子节点的深度为1，依此类推。对于非二叉树，每个节点可以拥有任意数量的子节点，这使得计算特定节点的深度成为一项常见的任务。
我们的目标是设计一个方法，给定一个树中的节点，能够准确返回其相对于根节点的深度。树中的每个节点至少包含两个属性：一个用于标识节点的 name 和一个存储其直接子节点的数组 children。由于树的天然递归特性，递归算法是解决此类问题的理想选择。
核心实现思路：递归遍历
要确定一个特定节点的深度，最关键的思路是从树的根节点开始进行遍历。当我们从根节点向下遍历时，每深入一层，深度值就增加1。当找到目标节点时，当前累积的深度值即为该节点的深度。如果遍历完所有子树仍未找到目标节点，则表示该节点不在当前搜索路径中。
我们将探讨两种主要的递归实现方案。
方案一：从根节点向下搜索目标节点
这种方法最为直观，它将深度计算逻辑封装在树的节点类中，并从根节点发起调用，传入目标节点的名称进行搜索。
立即学习“Java免费学习笔记（深入）”；
方法描述
getDepth(targetName) 方法被定义在 Node 类上，并由树的根节点调用。它接收一个 targetName 参数，表示我们想要查找其深度的目标节点的名称。
递归逻辑
-
基线条件 (Base Case):
- 如果当前节点的 name 与 targetName 匹配，说明我们找到了目标节点。此时，该节点的深度就是0（相对于当前搜索起点）。
- 如果当前节点没有子节点，并且它也不是目标节点，那么目标节点不在当前路径中，返回一个表示“未找到”的值（例如 -1）。
-
递归步骤 (Recursive Step):
- 遍历当前节点的所有子节点。
- 对每个子节点递归调用 child.getDepth(targetName)。
- 如果子节点返回的深度值大于或等于0（表示在子树中找到了目标节点），那么目标节点的实际深度就是子节点返回的深度值加1（因为当前节点比其子节点高一层）。
- 如果遍历完所有子节点都没有找到目标节点，则返回 -1。
示例代码
class Node { constructor(name, ...children) { this.name = name; this.children = children; // children是一个数组，可以包含任意数量的子节点 } /** * 从当前节点开始，向下搜索指定名称的节点，并返回其深度。 * 如果当前节点就是目标节点，深度为0。 * 如果未找到，返回-1。 * @param {string} targetName - 目标节点的名称 * @returns {number} - 目标节点的深度，如果未找到则为-1 */ getDepth(targetName) { // 基线条件1：如果当前节点就是目标节点，其深度为0 if (this.name === targetName) { return 0; } // 递归步骤：遍历所有子节点 for (const child of this.children) { // 递归调用子节点的getDepth方法 const depthInChildSubtree = child.getDepth(targetName); // 如果在子树中找到了目标节点（返回值 >= 0） if (depthInChildSubtree >= 0) { // 目标节点的深度是子树深度加1 return depthInChildSubtree + 1; } } // 基线条件2：遍历完所有子节点仍未找到，返回-1 return -1; } } // 构建示例树结构 const root = new Node("root", new Node("A", new Node("C", new Node("G"), new Node("H"), new Node("I") ), new Node("D") ), new Node("B", new Node("E"), new Node("F") ) ); // 示例调用：计算节点"C"的深度 const depthOfC = root.getDepth("C"); console.log(`节点 "C" 的深度: ${depthOfC}`); // 预期输出: 2 // 示例调用：计算节点"G"的深度 const depthOfG = root.getDepth("G"); console.log(`节点 "G" 的深度: ${depthOfG}`); // 预期输出: 3 // 示例调用：计算不存在的节点"X"的深度 const depthOfX = root.getDepth("X"); console.log(`节点 "X" 的深度: ${depthOfX}`); // 预期输出: -1
注意事项
- 此方法要求从树的根节点开始调用，并传入目标节点的名称。
- 返回值为 -1 表示在整个树中未找到指定名称的节点。
- 这种方法是最常用且最直观的，因为它模拟了我们从根部向下探索树的过程。
方案二：从目标节点向上追溯至根节点（通过根节点参数）
第二种方法是另一种思维方式，它将深度计算方法定义在 Node 类上，但由目标节点自身调用，并传入树的根节点作为参数。
方法描述
getDepthWithRespectTo(root) 方法由我们想要获取深度的 目标节点 调用，并传入整个树的 root 节点。其核心思想是，如果当前节点就是根节点，则深度为0；否则，在根节点的子节点中递归查找当前节点。
递归逻辑
-
基线条件 (Base Case):
- 如果当前节点 (this) 就是传入的 root 节点，那么它的深度就是0。
-
递归步骤 (Recursive Step):
- 遍历 root 节点的所有子节点。
- 对每个子节点 parent，递归调用 this.getDepthWithRespectTo(parent)。这里 this 始终是目标节点。
- 如果递归调用返回的深度值大于或等于0（表示目标节点在某个子树中被找到），那么目标节点的实际深度就是子树深度加1。
- 如果遍历完 root 的所有子节点都没有找到目标节点，则返回 -1。
示例代码
class Node { constructor(name, ...children) { this.name = name; this.children = children; } /** * 计算当前节点相对于给定根节点的深度。 * 如果当前节点就是根节点，深度为0。 * 如果当前节点不在以给定根节点为起点的树中，返回-1。 * @param {Node} root - 树的根节点 * @returns {number} - 当前节点的深度，如果未找到则为-1 */ getDepthWithRespectTo(root) { // 基线条件1：如果当前节点就是传入的根节点，深度为0 if (this === root) { return 0; } // 递归步骤：在根节点的子节点中查找当前节点 for (const child of root.children) { // 递归调用，尝试在child为根的子树中找到当前节点 const depth = this.getDepthWithRespectTo(child); // 如果在子树中找到了当前节点（返回值 >= 0） if (depth >= 0) { // 当前节点的深度是子树深度加1 return depth + 1; } } // 基线条件2：遍历完所有子节点仍未找到，返回-1 return -1; } } // 构建示例树结构，并保留对节点"C"的引用 let nodeC; // 用于保存对节点C的引用 const root = new Node("root", new Node("A", nodeC = new Node("C", // 节点C被赋值给nodeC new Node("G"), new Node("H"), new Node("I") ), new Node("D") ), new Node("B", new Node("E"), new Node("F") ) ); // 示例调用：计算节点"C"相对于根节点的深度 const depthOfC_alt = nodeC.getDepthWithRespectTo(root); console.log(`节点 "C" 的深度 (方案二): ${depthOfC_alt}`); // 预期输出: 2 // 假设我们有一个不在树中的节点 const nodeX = new Node("X"); const depthOfX_alt = nodeX.getDepthWithRespectTo(root); console.log(`节点 "X" 的深度 (方案二): ${depthOfX_alt}`); // 预期输出: -1
注意事项
- 此方法需要你已经拥有目标节点的引用，并将其作为 this 调用方法，同时传入树的根节点。
- 从逻辑上讲，它是在根节点的子树中查找当前调用方法的节点。
- 虽然功能上等价于方案一，但在实际应用中，方案一（从根向下搜索指定名称）可能更常见和直观，因为它更符合我们查找一个已知名称节点的思维模式。
总结与最佳实践
本文介绍了两种在JavaScript中计算非二叉树节点深度的递归方法。
-
方案一：root.getDepth(targetName)
- 从树的根节点开始向下遍历，查找指定名称的节点。
- 逻辑清晰，易于理解和实现。
- 适用于已知目标节点名称，需要从根节点开始搜索的场景。
-
方案二：targetNode.getDepthWithRespectTo(root)
- 由目标节点调用，传入根节点，在根的子树中递归查找自身。
- 适用于已经持有目标节点引用，并需要计算其相对于特定根节点深度的场景。
选择哪种方案取决于具体的应用场景和偏好。 在大多数情况下，方案一由于其直观的搜索逻辑和通过名称查找的便利性，会是更常用的选择。两种方案都利用了递归的强大功能，以简洁高效的方式解决了树形结构的深度计算问题。
在实现时，务必注意递归的基线条件和返回值，特别是当目标节点未找到时的处理，通常返回一个负值（如-1）来表示。这有助于避免无限递归并正确处理异常情况。理解并熟练运用递归是处理树形数据结构的关键技能之一。