LeetCode刷题037 从上到下打印二叉树

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

题目

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

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

    3
   / \
  9  20
    /  \
   15   7
返回：

[3,9,20,15,7]

提示：

节点总数 <= 1000

题解

二叉树的层次遍历

Code

/**
 * 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> list = new ArrayList<>();
        ArrayDeque<TreeNode> q = new ArrayDeque<>();
        TreeNode p = null;
        q.add(root);
        while(!q.isEmpty()){
            p = q.poll();
            list.add(p.val);
            if(p.left != null){
                q.add(p.left);
            }
            if(p.right != null){
                q.add(p.right);
            }
        }
        
        return list.stream().mapToInt(x -> (int) x).toArray();
    
    }
}