29-剑指 Offer 32 - I. 从上到下打印二叉树

思而不学则殆m

于 2022-12-31 15:46:35 发布

阅读量77

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指Offer(第2版) 文章标签：数据结构

本文链接：https://blog.youkuaiyun.com/WWXDwrn/article/details/128506018

剑指Offer(第2版) 专栏收录该内容

41 篇文章

订阅专栏

该博客介绍了一种使用队列进行迭代的方法来实现二叉树的层序遍历，具体步骤包括创建队列，将根节点入队，然后在循环中逐层取出节点并添加其值到结果列表中，同时将其左右子节点入队，直到队列为空。最终将结果列表转换为数组返回。这种方法适用于节点总数小于等于1000的二叉树。

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

题目

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
9 20
    / \
   15   7

返回：[3,9,20,15,7]

提示：节点总数 <= 1000

思路：队列，迭代，层序遍历

代码

/**
 * 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];
        }

        Queue<TreeNode> queue = new LinkedList<>();
        List<Integer> res = new ArrayList<>();

        queue.offer(root);
        while(!queue.isEmpty()) {
            TreeNode cur = queue.poll();
            res.add(cur.val);
            if(cur.left != null) {
                queue.offer(cur.left);
            }
            if(cur.right != null) {
                queue.offer(cur.right);
            }
        }

        int[] arr = new int[res.size()];
        for(int i = 0; i < res.size(); i++) {
            arr[i] = res.get(i);
        }

        return arr;
    }
}