问题描述:
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]
]
这道题大致的意思就是遍历每一层的节点,把同一层的节点存入一个list中,然后最后把所有的list存入一个总的list中。
这道题挺简单的,就是考二叉树的广度遍历。
一提到广度遍历,就不得不提到队列这个数据结构。这道题的主要想法就是将某一层的所有节点添加进一个空的队列,然后使用循环将他们逐个弹出(这里可以通过获取队列的大小(即那一层所有节点的数目)来限制循环的次数),并将他们的左右节点添加进队列。当添加进完所有左右节点时,下一层的所有节点就已经在队列中了。一直循环,直到最后队列为空,说明没有下一层的节点了,广度遍历结束。
直接上代码吧:
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
new LinkedList<List<Integer>>();
//储存最终结果
List<List<Integer>> result = new LinkedList<>();
//储存每一层的值
List<Integer> tempList = new LinkedList<>();
//储存节点
Queue<TreeNode> tempQueue = new LinkedList<>();
//添加根节点
tempQueue.offer(root);
while(tempQueue.peek() != null) {
//获取这一层的节点数
int levelSize = tempQueue.size();
//然后遍历这一层的所有节点
for(int i = 0; i < levelSize; i++) {
TreeNode tempNode = tempQueue.poll();
tempList.add(tempNode.val);
//将左右节点添加进队列中
if(tempNode.left != null)
tempQueue.offer(tempNode.left);
if(tempNode.right != null)
tempQueue.offer(tempNode.right);
}
//将tempList添加进result中
result.add(new LinkedList<Integer>(tempList));
//清空tempList中的节点
tempList.clear();
}
return result;
}
谢谢大家观看我的博客。如果有不明白的地方,或者是文中有错误的地方,欢迎指出,谢谢!