java广度遍历二叉树

原创

mob64ca12dc88a3 2023-10-06 14:03:29 ©著作权

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

广度优先遍历二叉树的实现方法

简介

在二叉树数据结构中，广度优先遍历（Breadth-First Traversal，简称BFS）是一种通过层序遍历的方式遍历二叉树的方法。它从树的根节点开始，逐层遍历每个节点，直到遍历完整个二叉树。本文将介绍如何使用Java语言实现二叉树的广度优先遍历。

流程图

下面是广度优先遍历二叉树的流程图：

graph TD
    A(开始)
    B[初始化队列，将根节点入队]
    C[判断队列是否为空]
    D[从队列中取出一个节点]
    E[访问当前节点]
    F[将当前节点的左右子节点入队]
    G[返回到步骤C]
    H(结束)
    A --> B --> C
    C --> D
    D --> E
    E --> F
    F --> G
    G --> C
    C --> H

实现步骤

以下是实现广度优先遍历二叉树的具体步骤：

步骤1：初始化队列

首先，我们需要创建一个队列（可以使用Java中的LinkedList类实现）。然后，将二叉树的根节点入队。

Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);

步骤2：判断队列是否为空

在每一轮遍历之前，我们需要判断队列是否为空。如果队列为空，说明已经遍历完整个二叉树，可以结束遍历。

while (!queue.isEmpty()) {
    // 遍历当前层的节点
}

步骤3：取出一个节点并访问

从队列中取出一个节点，然后访问该节点。访问可以是打印节点的值、将节点的值加入到结果列表等等。

TreeNode current = queue.poll();
System.out.println(current.val); // 访问当前节点的值

步骤4：将当前节点的左右子节点入队

将当前节点的左右子节点（如果存在）入队，以便在下一轮遍历中访问。

if (current.left != null) {
    queue.add(current.left);
}
if (current.right != null) {
    queue.add(current.right);
}

步骤5：返回到步骤2

重复步骤2至步骤4，直到队列为空，即遍历完整个二叉树。

代码实现

下面是完整的Java代码实现（假设二叉树的节点类为TreeNode，包含一个整型的值和左右子节点的引用）：

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

public class BinaryTreeBFS {
    public void breadthFirstTraversal(TreeNode root) {
        if (root == null) {
            return;
        }
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        
        while (!queue.isEmpty()) {
            TreeNode current = queue.poll();
            System.out.println(current.val); // 访问当前节点的值
            
            if (current.left != null) {
                queue.add(current.left);
            }
            
            if (current.right != null) {
                queue.add(current.right);
            }
        }
    }
}

以上代码实现了二叉树的广度优先遍历。你可以将根节点传入breadthFirstTraversal方法，即可输出二叉树的广度优先遍历结果。