题目:
给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
例如:
给定的二叉树是{3,9,20,#,#,15,7},
该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]
对于不分层直接输出的层次遍历,直接采用队列的思想,首先root入队,针对队列的每一个元素,当前元素出队,访问,当前元素的左右孩子入队,直到队列为空,这里需要进行分层输出,所以在进行访问的时候也需要进行分层处理。主要思想是,每一层采用一个队列,在对当前层遍历的同时,将当前层的孩子放入到新的队列,对当前层遍历结束后,对新的队列进行遍历。
完整代码如下:
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
List<TreeNode> nodes = new ArrayList<>();
nodes.add(root);
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
int count = 0;
while(nodes.size()!=0) {
List<TreeNode> temp = new ArrayList<>(); //构建新的队列
ArrayList<Integer> list = new ArrayList<>(); //当前层的遍历结果
for(int i=0;i<nodes.size();i++) {
TreeNode now = nodes.get(i);
if(now!=null) {
list.add(now.val);
temp.add(now.left);
temp.add(now.right);
}
}
nodes=temp;
if(list.size()>0){
res.add(list);
}
}
return res;
}