力扣打卡:199. 二叉树的右视图
解题思路
思路:层序遍历
-
因为层序遍历每一层最后一个节点都会没扫描到,所以从层序遍历入手
-
每一层的最后一个节点刚好就是层序遍历时,定义的
size-1
-
修改层序遍历的框架,如果是
size-1
那么就将值放进去
同样的思路:如果是从左边看呢,那么也就是层序遍历的第一个数
代码
/**
* 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> ans = new ArrayList<>();
if(root==null) return ans;
Queue<TreeNode> que = new LinkedList<>();
que.offer(root);
while(!que.isEmpty()){
int size = que.size();
for(int i=0; i<size; i++){
TreeNode p = que.poll();
if(i==size-1){
ans.add(p.val);
}
if(p.left!=null) que.offer(p.left);
if(p.right!=null) que.offer(p.right);
}
}
return ans;
}
}