思想:BFS遍历方法
用队列思想,存储每一层结点,第一层入队列,第二层出队列。
Queue a=new ArrayDeque<>();
判断root是否为空,不为空则入队列。
判断队列是否为空,不为空则执行循环。
定义每一层的大小n,引入一个list对象 level
出队元素存到level中。之后分别将左子树和右子树入队列。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
Queue<TreeNode> a=new ArrayDeque<>();
if(root!=null){
a.add(root);
}
while(!a.isEmpty()){
int n=a.size();
List<Integer> level=new ArrayList<>();
for(int i=0;i<n;i++) {
TreeNode node=a.poll();
level.add(node.val);
if(node.left!=null)
a.add(node.left);
if(node.right!=null)
a.add(node.right);
}
res.add(level);
}
return res;
}
}