二叉树的层序遍历
对多道二叉树层序遍历相关题目,主要是用了两种遍历方式,一种是递归方式,一种是使用队列。代码模板可以用于多道题的解答,在此用两道题目记录这两种思路。
学习参考自代码随想录对应章节
class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> levelOrderBottom(TreeNode root) {
checkRes(root,0);
return shift(res);
}
public void checkRes(TreeNode node, int deep){
if(node == null) return ;
deep++;
if(res.size()<deep){
List<Integer> list = new ArrayList();
res.add(list);
}
res.get(deep-1).add(node.val);
checkRes(node.left, deep);
checkRes(node.right, deep);
}
public List<List<Integer>> shift(List<List<Integer>> arrayList){
for(int i=0,j=arrayList.size()-1;i<j;i++,j--){
List<Integer> temp=arrayList.get(i);
arrayList.set(i,arrayList.get(j));
arrayList.set(j, temp);
}
return arrayList;
}
}
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<>();
Deque<TreeNode> que = new LinkedList<>();
if (root == null){
return res;
}
TreeNode node = root;
que.offerLast(node);
while(!que.isEmpty()){
int levelSize = que.size();
for(int i=0; i<levelSize; i++){
TreeNode poll = que.pollFirst();
if(poll.left != null){
que.offerLast(poll.left);
}
if(poll.right != null){
que.offerLast(poll.right);
}
if(i == levelSize-1){
res.add(poll.val);
}
}
}
return res;
}
}