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

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

题目描述

解题思路

层次遍历。

注意，ArrayList 转 int[] 数组，需要手动用 for 循环。

class Solution {
    public int[] levelOrder(TreeNode root) {
        if (root == null) return new int[0];

        ArrayList<Integer> res = new ArrayList<>();

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

        while (!queue.isEmpty()) {
            TreeNode currNode = queue.poll();
            //保存结果
            res.add(currNode.val);
            //左右孩子入队
            if (currNode.left != null) queue.offer(currNode.left);
            if (currNode.right != null) queue.offer(currNode.right);
        }
        
        int[] resOfArray = new int[res.size()];
        for (int i = 0; i < resOfArray.length; i++) {
            resOfArray[i] = res.get(i);
        }
        return resOfArray;
    }
}