leetcode题解（四十）：102. Binary Tree Level Order Traversal

最新推荐文章于 2024-06-25 11:36:41 发布

原创最新推荐文章于 2024-06-25 11:36:41 发布 · 124 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

73 篇文章

订阅专栏

本文介绍了一种使用队列实现的二叉树层序遍历算法，通过先进先出的原则，逐层访问二叉树节点，适用于数据结构与算法学习。

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

在这里插入图片描述
二叉树的层序遍历
按照输入输出先进先出的思路，这里用栈结构

public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        List<List<Integer>> wrapList = new LinkedList<List<Integer>>();
        
        if(root == null) return wrapList;
        
        queue.offer(root);
        while(!queue.isEmpty()){
        	//记录层次元素数量，比如第一层有一个元素，第二层两个,...也可能缺失
            int levelNum = queue.size();
            List<Integer> subList = new LinkedList<Integer>();
            for(int i=0; i<levelNum; i++) {
                if(queue.peek().left != null) queue.offer(queue.peek().left);
                if(queue.peek().right != null) queue.offer(queue.peek().right);
                subList.add(queue.poll().val);
            }
            wrapList.add(subList);
        }
        return wrapList;
    }
}