LeetCode刷题【剑指 Offer 32 – II】从上到下打印二叉树 II

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

 

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

 

提示：

1. 节点总数 <= 1000

注意：本题与主站 102 题相同：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/

在原来单纯的BFS的基础上JAVA BFS 3 连发（2）
这是上一题的代码，先再瞅一眼。我们就拿这份代码做下修改

class Solution {
    public int[] levelOrder(TreeNode root) {
        if (root == null){
            return new int[0];
        }
        int [] list = new int[1009];
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int idx = 0;
        while (queue.size()>0){
            TreeNode node = queue.poll();
            list[idx++] = node.val;
            if (node.left != null){
                queue.offer(node.left);
            }
            if (node.right != null){
                queue.offer(node.right);
            }
        }
        int[] res= new int[idx];
        System.arraycopy(list,0,res,0,idx);
        return res;
    }
}

修改前的的一顿分析

因为本题要返回的的一层一层的结构了，所以不能对于遍历的queue就不能一股脑的直接poll()了
嗯，那就一段一段的poll()。

注意里面的注释对queue.size()进行for循环是有问题，每次poll、offer之后，queue.size()都会变化

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if (null!=root)queue.offer(root);
        while (!queue.isEmpty()){
            int size = queue.size();
            List<Integer> line = new ArrayList<>();
//            对queue.size()进行for循环是不对的，每次poll、offer之后，queue.size()都会变化
//            for (int i = 0; i < queue.size(); i++)
            while (size-- > 0){
                if (null != queue.peek().left)queue.offer(queue.peek().left);
                if (null != queue.peek().right)queue.offer(queue.peek().right);
                line.add(queue.poll().val);

            }
            res.add(line);
        }
        return res;
    }
}

是不是风格看起来有点不一样，因为List<Integer>不用关心长度问题，直接往里塞就行了，如果还想要原来那样的话，也问题不大，再改改。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if (null!=root)queue.offer(root);
        while (!queue.isEmpty()){
            int size = queue.size();
            Integer[] line = new Integer[size];
            while (size-- > 0){
                if (null != queue.peek().left)queue.offer(queue.peek().left);
                if (null != queue.peek().right)queue.offer(queue.peek().right);
                line[line.length-size-1] = queue.poll().val;
            }
            res.add( Arrays.asList(line));
        }
        return res;
    }
}

BFS相关合集

BFS三连发1

BFS三连发2

BFS三连发3