剑指 Offer 32 - I. 从上到下打印二叉树（Java、BFS）

原题地址

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

题目描述

题目思路

• 由上到下，自左向右遍历，即为二叉树的广度优先搜索（BFS），可以由队列（先进先出）来实现。

代码实现

广度优先搜索（BFS）

class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root == null) return new int[0];
        Queue<TreeNode> queue = new LinkedList<>();
        ArrayList<Integer> list = new ArrayList<>();

        queue.add(root);
        while(!queue.isEmpty()) {
            TreeNode node = queue.poll();
            list.add(node.val);

            if(node.left != null) queue.add(node.left);

            if(node.right != null) queue.add(node.right); 
        }
		//JDk1.8可采用流将列表转换为数组
		/*
		int[] res = 
			list.stream().mapToInt(Integer::intValue).toArray();
		*/
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }

        return res;
    }
}