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

最新推荐文章于 2025-07-19 00:15:00 发布

原创最新推荐文章于 2025-07-19 00:15:00 发布 · 2.1k 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

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

15 篇文章

订阅专栏

这篇博客主要介绍如何用Java解决输出二叉树中所有节点所在层级的问题。文章展示了具体的代码实现，并给出了运行结果，包括从第一层到第四层的节点。此外，还提到了与之相关的其他树的层次遍历问题。

一、问题

              1
       2             3
         4         5  6
                 7

输出所有节点所在的层数

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

二、代码

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

    /**
     * 给所有节点标记level
     */
    public void setLevel(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 ++;
                nextLevelFirstNode = null;
            }

            head.setLevel(level);
            System.out.println(head.getId() + ":level=" + head.getLevel());

            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.setLevel(tree.getRoot());
    }
}

运行结果

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