树：分层输出所有的节点 (Java)

最新推荐文章于 2024-08-10 15:30:32 发布

原创最新推荐文章于 2024-08-10 15:30:32 发布 · 790 阅读

2 ·

CC 4.0 BY-SA版权

8.4 数据结构&算法专栏收录该内容

15 篇文章

订阅专栏

本文介绍如何使用Java分层输出树的所有节点。通过层次遍历，展示从第一层到最后一层的节点，例如：level:1 1, level:2 2 3, level:3 4 5 6, level:4 7。并提供了相关主题，如树的层数计算等。" 102694868,8386654,移动端面试常见问题解析,"['移动端开发', '响应式设计', 'Bootstrap', '事件处理', '框架对比']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、问题

              1
       2             3
         4         5  6
                 7

分层输出所有的节点

level:1 1
level:2 2 3
level:3 4 5 6
level:4 7

二、代码

Java

@Data
public class Node {
    private Integer id;
    private Node left;
    private Node right;
    private Integer level;

    public Node(Integer id) {
        this.id = id;
    }

    public Node(Integer id, Node left, Node right) {
        this.id = id;
        this.left = left;
        this.right = right;
    }
}



@Data
public class Tree {
    private Node root;

    /**
     * 建树 【V80】
     *
     *              1
     *       2             3
     *         4         5  6
     *                 7
     */
    public void build() {
        Node node7 = new Node(7);
        Node node6 = new Node(6);
        Node node5 = new Node(5, node7, null);
        Node node4 = new Node(4);
        Node node3 = new Node(3, node5, node6);
        Node node2 = new Node(2, null, node4);
        root = new Node(1, node2, node3);
    }

    /**
     * 分层打印
     */
    public void printByLevel(Node root) {
        if (root == null) {
            return;
        }

        int level = 0;
        Node nextLevelFirstNode = root;    //level下一层的第一个节点

        Queue<Node> queue = new LinkedList<>();
        queue.offer(root);

        while (!queue.isEmpty()) {
            Node head = queue.poll();

            //到下一层的第一个节点了，level才+1
            if (nextLevelFirstNode != null && Objects.equals(head.getId(), nextLevelFirstNode.getId())) {
                level ++;
                System.out.print("\nlevel:" + level + "   ");
                nextLevelFirstNode = null;
            }

            System.out.print(head.getId()  + "  ");

            if (head.getLeft() != null) {
                queue.offer(head.getLeft());

                if (nextLevelFirstNode == null) {
                    nextLevelFirstNode = head.getLeft();
                }
            }

            if (head.getRight() != null) {
                queue.offer(head.getRight());

                if (nextLevelFirstNode == null) {
                    nextLevelFirstNode = head.getRight();
                }
            }
        }
    }

    public static void main(String[] args) {
        Tree tree = new Tree();
        tree.build();
        tree.printByLevel(tree.getRoot());
    }
}

运行结果

level:1 1
level:2 2 3
level:3 4 5 6
level:4 7