java 获取子节点到根节点的路径

原创

mob64ca12df5e97 2024-09-13 07:07:59 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12df5e97的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java 获取子节点到根节点的路径

在实际开发中，树形结构的数据常常需要处理。在这篇文章中，我们将讨论如何在Java中获取从一个指定子节点到根节点的路径，并通过具体的步骤、代码示例以及图示来帮助你更好地理解。

整体流程

在实现这个功能之前，我们首先需要明确每个步骤。以下是一张表格，展示了解决问题的流程。

步骤	描述
1	定义树的节点结构
2	构建树的结构
3	编写查找路径的方法
4	打印从子节点到根节点的路径

接下来，我们将详细讨论每一个步骤。

步骤详解

1. 定义树的节点结构

首先，我们需要定义一个表示树节点的类。每个节点将包含自己及其子节点的引用。以下是TreeNode类的实现：

// 定义树的节点
class TreeNode {
    int value; // 节点的值
    List<TreeNode> children; // 存储子节点的列表

    // 构造函数
    public TreeNode(int value) {
        this.value = value;
        this.children = new ArrayList<>(); // 初始化子节点列表
    }
}

2. 构建树的结构

接下来，我们需要构建一个树的实例。可以通过创建节点并将其绑定为子节点来构建树。在代码中，我们将添加几个节点，并形成如下所示的树：

以下是构建这棵树的代码：

public class TreeDemo {
    public static void main(String[] args) {
        // 创建节点
        TreeNode root = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        TreeNode node3 = new TreeNode(3);
        TreeNode node4 = new TreeNode(4);
        TreeNode node5 = new TreeNode(5);
        TreeNode node6 = new TreeNode(6);
        
        // 建立树的结构
        root.children.add(node2);
        root.children.add(node3);
        root.children.add(node4);
        node2.children.add(node5);
        node2.children.add(node6);
    }
}

3. 编写查找路径的方法

为了找到从子节点到根节点的路径，我们需要一个递归的方法来遍历树，并在找到目标节点时记录路径。以下是实现这个方法的代码：

import java.util.ArrayList;
import java.util.List;

public class TreeDemo {
    // 存储路径的列表
    public static List<Integer> path = new ArrayList<>();
    
    // 查找路径的递归方法
    public static boolean findPath(TreeNode root, int target) {
        // 如果当前节点为空，返回false
        if (root == null) {
            return false;
        }

        // 将当前节点的值添加到路径中
        path.add(root.value);

        // 如果当前节点是目标节点，返回true
        if (root.value == target) {
            return true;
        }

        // 遍历当前节点的子节点
        for (TreeNode child : root.children) {
            if (findPath(child, target)) {
                return true; // 如果在子节点中找到路径，返回true
            }
        }

        // 如果当前节点不是目标节点，将其从路径中移除
        path.remove(path.size() - 1);
        return false; // 返回false表示未找到
    }
}

4. 打印从子节点到根节点的路径

最后，我们可以在主方法中调用查找路径的方法，并输出从指定子节点到根节点的路径：

public class TreeDemo {
    public static void main(String[] args) {
        // 创建节点并建立树的结构...
        
        // 指定要查找的目标节点
        int target = 6;

        // 调用查找路径的方法
        if (findPath(root, target)) {
            System.out.print("Path from node " + target + " to root: ");
            // 逆序输出路径
            for (int i = path.size() - 1; i >= 0; i--) {
                System.out.print(path.get(i) + " ");
            }
        } else {
            System.out.println("Node " + target + " not found in the tree.");
        }
    }
}

关系图

下面是树形结构的关系图，使用mermaid的ER图语法表示：

erDiagram
    TreeNode {
        int value
        List<TreeNode> children
    }
    TreeNode ||--o{ TreeNode : has

流程图

使用mermaid的流程图语法，展示出获取路径的流程：

flowchart TD
    A[开始] --> B[创建TreeNode]
    B --> C[建立节点关系]
    C --> D[调用findPath方法]
    D --> E{节点是否为空}
    E --Yes--> F[路径加入当前节点]
    E --No--> G{当前节点是否为目标节点}
    G --Yes--> H[返回true]
    G --No--> I[遍历子节点]
    I --> E
    F --> J[从路径中移除当前节点]
    J --> K[返回false]
    H --> L[打印路径]
    L --> M[结束]