二叉树的层序遍历
思路
前面二叉树深度遍历采用了递归,迭代(利用栈)的方法,二叉树层序遍历则是需要使用队列来完成。
设置内外两层循环(内循环负责记录每层的节点),利用上次循环添加的节点的个数来控制内层循环次数(即上次循环结束时的queue.size),内循环负责出队并记录该节点值,然后入队其左右节点。
102.二叉树的层序遍历
107.二叉树的层序遍历2:自底向上层序遍历同理,只需最后将结果逆置即可,使用栈来辅助完成。
199.二叉树的右视图 :做法同上,只需将每层内循环的List最后一个值取出即可。
637.二叉树的平均值 :将每层循环List值取出求平均值。
429.N叉树的层序遍历:修改判断左右子节点是否为空的部分,使遍历list中的每个节点元素即可,其余相同
515.在每个树行中找最大值:利用一个变量记录每个循环中数值的最大值(可用Math.max函数),内循环每结束一次加入至结果list。
116.填充每个节点的下一个右侧节点指针:可设置一个List利用队列来记录每一层的节点,之后遍历List将前size-1的节点指向其后一个节点,最后一个节点不动。
117.填充每个节点的下一个右侧节点指针2:同116层序遍历解法
104.二叉树的最大深度:可以用层次遍历,最后返回list的size。
111.二叉树最小深度:层次遍历,遇到第一个叶子节点返回。
总结
层次遍历解法:
1.利用队列完成二叉树层序遍历
2.需要利用size来记录下一层节点的个数
3.内循环使用List来保存节点值,最后存入List
上述题目用层次遍历解普遍属于简单难度,部分题目可以使用递归解法,如116.117.104.111.明天尝试对这四道题目进行递归解法。
代码
代码篇幅原因贴第一个代码。
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
if (root==null) return res;
Queue<TreeNode> queue=new ArrayDeque<>();
queue.offer(root);
while (!queue.isEmpty()){
int size=queue.size();
List<Integer> inner=new ArrayList<>();
while (size!=0){
TreeNode temp=queue.poll();
if (temp!=null){
inner.add(temp.val);
if (temp.left!=null) queue.offer(temp.left);
if (temp.right!=null) queue.offer(temp.right);
}
size--;
}
res.add(inner);
}
return res;
}