java遍历树获取叶子节点数据

原创

mob64ca12d9081f 2023-10-01 04:04:49 ©著作权

文章标签 子节点数据中序遍历 文章分类 Java 后端开发

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

Java遍历树获取叶子节点数据

树结构是一种常见的数据结构，它由节点组成，每个节点可以有零个或多个子节点。在实际开发中，经常需要遍历树来获取叶子节点的数据。

什么是叶子节点？

在树结构中，叶子节点是指没有子节点的节点。可以将树看作是由根节点和叶子节点组成的。叶子节点是树的末端节点，没有后续节点。

遍历树的方法

遍历树的方法有多种，其中包括前序遍历、中序遍历和后序遍历。这里我们使用中序遍历来获取叶子节点的数据。

中序遍历

中序遍历是一种深度优先遍历方法，它的遍历顺序是先遍历左子树，然后访问根节点，最后遍历右子树。下面是中序遍历的示意图：

graph LR
A --> B
A --> C
B --> D
B --> E
C --> F

中序遍历的遍历顺序为：D -> B -> E -> A -> F -> C。

代码示例

下面是一个使用Java代码实现的遍历树获取叶子节点数据的示例：

public class TreeNode {
    private int data;
    private TreeNode left;
    private TreeNode right;

    public TreeNode(int data) {
        this.data = data;
    }

    public void setLeft(TreeNode left) {
        this.left = left;
    }

    public void setRight(TreeNode right) {
        this.right = right;
    }

    public boolean isLeaf() {
        return left == null && right == null;
    }

    public List<Integer> getLeafNodes() {
        List<Integer> leafNodes = new ArrayList<>();
        getLeafNodesHelper(this, leafNodes);
        return leafNodes;
    }

    private void getLeafNodesHelper(TreeNode node, List<Integer> leafNodes) {
        if (node == null) {
            return;
        }

        if (node.isLeaf()) {
            leafNodes.add(node.data);
        }

        getLeafNodesHelper(node.left, leafNodes);
        getLeafNodesHelper(node.right, leafNodes);
    }
}

上面的代码中，我们定义了一个TreeNode类来表示树节点，其中包含了节点的数据、左子节点和右子节点。isLeaf()方法用来判断节点是否为叶子节点，getLeafNodes()方法用来获取树的所有叶子节点的数据。

下面是一个使用示例：

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);

    root.setLeft(node2);
    root.setRight(node3);
    node2.setLeft(node4);
    node2.setRight(node5);

    List<Integer> leafNodes = root.getLeafNodes();
    for (int data : leafNodes) {
        System.out.println(data);
    }
}

运行上面的代码，输出结果为：4和5，这两个数字是树的叶子节点的数据。