Java层次遍历

概述

在进行树结构的遍历时,层次遍历是一种常用的方法。层次遍历可以按照树的层次逐层遍历节点,从根节点开始,依次遍历每一层的节点。这种遍历方式可以通过使用队列来实现,具体的流程如下表所示:

步骤 操作
1 创建一个空队列,并将根节点入队列
2 当队列不为空时,执行以下操作:<br> * 从队列中取出一个节点,访问该节点的值<br> * 若该节点有左子节点,将左子节点入队列<br> * 若该节点有右子节点,将右子节点入队列
3 重复步骤2,直到队列为空

代码实现

以下是实现Java层次遍历的代码实现,注释中解释了每一行代码的作用。

import java.util.LinkedList;
import java.util.Queue;

public class LevelOrderTraversal {
    public void levelOrderTraversal(TreeNode root) {
        // 创建一个空队列
        Queue<TreeNode> queue = new LinkedList<>();
        // 将根节点入队列
        queue.offer(root);
        
        while (!queue.isEmpty()) {
            // 从队列中取出一个节点
            TreeNode node = queue.poll();
            
            // 访问该节点的值
            System.out.println(node.val);
            
            // 若该节点有左子节点,将左子节点入队列
            if (node.left != null) {
                queue.offer(node.left);
            }
            
            // 若该节点有右子节点,将右子节点入队列
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
    }
}

在上述代码中,我们首先创建了一个空队列 queue,然后将根节点 root 入队列。接下来,我们使用一个循环来执行步骤2,直到队列为空。在每次循环中,我们从队列中取出一个节点,并通过访问 node.val 来获取该节点的值。然后,我们检查该节点是否有左子节点和右子节点,如果有,则将它们分别入队列。

示例

假设我们有以下的二叉树结构:

      1
    /   \
   2     3
  / \   / \
 4   5 6   7

我们可以按照层次遍历的方式输出节点的值:

public class Main {
    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(7);
        
        LevelOrderTraversal lot = new LevelOrderTraversal();
        lot.levelOrderTraversal(root);
    }
}

输出结果将是:

1
2
3
4
5
6
7

总结

通过使用队列,我们可以很方便地实现Java层次遍历。层次遍历是一种按照树的层次逐层遍历节点的方法,需要注意的是要在遍历过程中,按照节点的顺序将它们的子节点入队列。这种遍历方式通常用于树结构中需要按照层次进行处理的场景。