二叉树的层序遍历

二叉树的层序遍历通常使用队列(Queue)来实现。层序遍历也称为广度优先遍历(Breadth-First Traversal),它按照树的层级,从上到下、从左到右遍历节点。

java 复制代码
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new LinkedList<>();
        if (root==null){
            return ans;
        }
        //构建一个队列 先进先出
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()){
            List<Integer> list = new ArrayList<>();
            //每次记录 queue的长度
            int size = queue.size();
            for (int i = 0; i <size ; i++) {
                TreeNode cur = queue.poll();
                if (cur.left!=null){
                    queue.add(cur.left);
                }
                if (cur.right!=null){
                    queue.add(cur.right);
                }
                list.add(cur.val);
            }
            ans.add(list);
        }
        return ans;
    }

这个 levelOrder 方法首先检查根节点是否为空,如果为空则直接返回一个空的列表。然后,它创建一个队列并将根节点入队。在循环中,它首先获取当前层的节点数(即队列的大小),然后遍历这些节点,将它们的值添加到当前层的列表中,并将它们的子节点(如果存在)入队。完成当前层的遍历后,将当前层的列表添加到结果列表中。这个过程一直持续到队列为空,即遍历完所有层。最后,返回结果列表。

java 复制代码
public static void main(String[] args) {  
    // 创建一个二叉树  
    TreeNode root = new TreeNode(3);  
    root.left = new TreeNode(9);  
    root.right = new TreeNode(20);  
    root.right.left = new TreeNode(15);  
    root.right.right = new TreeNode(7);  
      
    BinaryTreeLevelOrderTraversal solution = new BinaryTreeLevelOrderTraversal();  
    List<List<Integer>> result = solution.levelOrder(root);  
      
    // 打印层序遍历的结果  
    for (List<Integer> level : result) {  
        System.out.println(level);  
    }  
}

// 输出结果
[3]  
[9, 20]  
[15, 7]
相关推荐
Dream it possible!7 分钟前
蓝桥杯_翻转_C++
c++·算法·蓝桥杯
Ivanqhz7 分钟前
BURG(自底向上重写生成器)
服务器·数据库·人工智能·深度学习·算法
言乐614 分钟前
逻辑回归与利弊
人工智能·算法·机器学习·数据挖掘·逻辑回归
海绵宝宝转agent22 分钟前
Leetcode100 二叉树的中序遍历
java·算法
根目录下的猫28 分钟前
ROS 2 C++(rclcpp)版本,实现话题订阅、话题发布,以及完整编译运行流程
开发语言·c++·算法
大侠归来2 小时前
C 语言 | 在函数中操作数组
c语言·c++·算法
怕浪猫7 小时前
GEO 优化到底是什么?AI时代内容创作者必须懂的新技能
算法·面试·github
朝朝辞暮i8 小时前
C++ 第 23 课:class —— 开始真正进入面向对象
开发语言·c++·算法
7yewh9 小时前
SLAM 三维空间刚体运动(2)
数据结构·人工智能·机器人·嵌入式·slam
Evand J9 小时前
【MATLAB例程】三维RRT+APF避障路径规划与到达角(AOA)定位算法|三维路径优化与定位仿真例程
算法·matlab·路径规划·代码·定位·rrt·aoa