102. Binary Tree Level Order Traversal (二叉树层次遍历)

题目

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

return its level order traversal as:

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

分析

1 结果的二维数组的展现方式 想到 linklist的嵌套linklist

2 中序遍历用到的数据结构是栈

   层次遍历用到的数据结构是队列，保证按层输出，每一个循环都记下这一层的结果，并且把下一层的结点加入

3 queue 的特点 先进先出，因此和适合层次遍历的这种情况

               queue.offer 在队列的末尾加入一个元素

               queue.peek 取出队列的头的元素值，注意这时候并不弹出这个值

代码

public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> wrapList = new LinkedList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(root == null) return wrapList;
        queue.offer(root);
        while (!queue.isEmpty()){
            int levelNum = queue.size();
            List<Integer> subList = new LinkedList<>();
            for (int i = 0; i < levelNum; i++) {
                if(queue.peek().left != null) queue.offer(queue.peek().left);
                if(queue.peek().right != null) queue.offer(queue.peek().right);
                subList.add(queue.poll().val);
            }
            wrapList.add(subList);
        }
        return wrapList;   
    }