题目:给定一个二叉树的 根节点 root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
根据题目,这道题需要获取的是,此二叉树中每一层从左到右数最后一个节点的值。
要获取每一层的值,直接对该二叉树进行广度优先遍历。
即创建一个队列Queue,将二叉树放入队列中,此时可以直接从队列中获取到根节点的值。
Deque<TreeNode> queue = new ArrayDeque<TreeNode>();
queue.offer(root);
之后,我们从队列中取出节点,然后把子节点放入队列
在之后,继续从队列中把之前放入的子节点都取出,再放入子节点的子节点...
最后取出子节点的子节点的子节点的......子节点的子节点,放入出子节点的子节点的子节点的......子节点的子节的子节点........
每一次取出,都把取出的节点值放在一个列表中,记录为这颗二叉树这一层中的值,然后将最后一个值放入最后要作为答案返回的列表中。
代码:(就是模板化的广度优先层序遍历而已,基本就同【LeetCode学习记录102】——二叉树的层序遍历-CSDN博客)
public List<Integer> rightSideView(TreeNode root) {
List<Integer> ret = new ArrayList<Integer>();
if(root == null) return ret;
Deque<TreeNode> queue = new ArrayDeque<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()){
List<Integer> tmp = new ArrayList<Integer>();
int size = queue.size();
if(size>0) {
for (int i = 0; i < size; i++)
{
TreeNode node = queue.poll();
tmp.add(node.val);
if(node.left!= null)
{
queue.offer(node.left);
};
if(node.right!= null) {
queue.offer(node.right);
};
}
System.out.println(tmp);
ret.add(tmp.get(tmp.size()-1));
}
}
return ret;
}