题目
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;
}