【剑指 Offer 32 - I. 从上到下打印二叉树】（二叉树的层次遍历）

原创于 2022-09-28 21:42:05 发布 · 568 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #职场和发展

JAVA 专栏收录该内容

126 篇文章

订阅专栏

本文介绍了一种实现二叉树层次遍历的方法，通过使用队列数据结构来逐层访问二叉树的节点。首先将根节点放入队列中，然后依次访问队列中的每个节点，并将它们的子节点加入队列，直至队列为空。

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

【解题思路】

二叉树的层次遍历，需要新建一个队列，第一次将根节点放进队列中，如果队列不为空，依次访问队列中的元素，访问时，如果该元素的左右节点不为空，将左右节点插入队列末尾，该元素出队。依次访问队列中的元素，直到队列为空。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root == null) return new int[0];

        ArrayList<Integer> ans = new ArrayList<Integer>();
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);

        while(queue.isEmpty() == false)
        {
            TreeNode p = queue.poll();
            ans.add(p.val);
            if(p.left != null)
            {
                queue.offer(p.left);
            }
            if(p.right != null)
            {
                queue.offer(p.right);
            }
        }

        return ans.stream().filter(Integer -> Integer!= null).mapToInt(i -> i).toArray();
    }
}