剑指Offer 32 - Ⅰ.从上到下打印二叉树Java实现

近期刚好刷到这道题，刚好写写思路

 

 对于此题中二叉树的遍历使用广度优先搜索，即BFS，利用队列先进先出的特点进行实现。

流程思路

1、如果根节点为空，则直接返回一个空数组

2、初始化队列，并将根节点放入队列中，并新建一个List存放结果

3、对树进行遍历，当队列为空后结束循环

        1、从队列中取出首个元素

        2、将所取出的节点的值存入List

        3、如果该节点的子节点不为空，则将子节点放入队列中（先左后右）

4、返回结果，因为本题有要求返回一个int数组，所以需要将集合转化为数组

代码如下：

/**
 * 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];
        ArrayDeque<TreeNode> queue = new ArrayDeque<>();
        List<Integer> res = new ArrayList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            res.add(node.val);
            if(node.left != null) queue.offer(node.left);
            if(node.right != null) queue.offer(node.right);
        }
        int[] ans = new int[res.size()];
        for(int i = 0 ; i < ans.length ; i++){
            ans[i] = res.get(i);
        }
        return ans;
     }
}

 此题思路参考自Krahets大佬，但在实现过程中，没有使用LinkedList，而是改用了ArrayDeque。

ArrayDeque是 Deque的实现类，可以作为栈来使用，效率高于 Stack；也可以作为队列来使用，效率高于 LinkedList，其是Java中双端队列的数组实现，而LinkedList则是双端队列的链表实现，所以具体的选择要视情况而定，不过在此题中ArrayDeque的性能更优。