题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路
采用队列
一开始将头节点放在队列里面;
然后进入一个while循环,
- 获取此时队列的大小k,k一定是某一层节点的个数
- 循环取出k个节点,这k个节点为其中的一层
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root==null){
return new ArrayList<>();
}
Queue<TreeNode>queue=new LinkedList<>();
List<List<Integer>>res=new ArrayList<>();
queue.add(root);
while(!queue.isEmpty()){
int k=queue.size();
List<Integer>tmp=new ArrayList<>();
for(int i=0;i<k;i++){
TreeNode t=queue.poll();
tmp.add(t.val);
if(t.left!=null)queue.add(t.left);
if(t.right!=null)queue.add(t.right);
}
res.add(tmp);
}
return res;
}
}