题目描述:
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路:
直接使用一个队列将二叉树中的节点按层次遍历存储,每次弹出同一层节点的时候将所有节点的数值存储到一个List
中,将该List放到List<List<Integer>>中,然后存储下一层的所有节点到队列中,不断循环直至队列为空。
实现代码:
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)
return new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
List<List<Integer>> res = new ArrayList<>();
while(!queue.isEmpty()){
ArrayList<Integer> list = new ArrayList<>();
for(int i = queue.size(); i > 0; i--){//遍历每一层节点
TreeNode temp = queue.poll();
list.add(temp.val);
if(temp.left != null)
queue.offer(temp.left);
if(temp.right != null)
queue.offer(temp.right);
}
res.add(list);
}
return res;
}