剑指 Offer 32 - I. 从上到下打印二叉树
从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回：
[3,9,20,15,7]
提示：
节点总数 <= 1000
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof/
解题
/** * 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) { //如果等于null，就直接返回一个空的int[]数组 if(root==null) return new int[]{}; //利用队列先进先出的特性，保存每一个根节点 Queue<TreeNode> queue = new LinkedList<>(); //存储数据 List<Integer> list = new ArrayList<>(); //先把根入队 queue.offer(root); //队列不为空的时候，一直执行 while (!queue.isEmpty()){ //第一步：取出根元素,存储到list中 list.add(queue.peek().val); //第二步：弹出队列中的根元素 TreeNode pollRoot = queue.poll(); //第三步：分别把根的左子树和右子树入队 if(pollRoot.left!=null) queue.offer(pollRoot.left); if(pollRoot.right!=null) queue.offer(pollRoot.right); } // 将list转为int数组返回 return list.stream().mapToInt(Integer::intValue).toArray(); } }
