1. 本题知识点
二叉树、广度优先遍历
2. 题目描述
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
3. 解题思路
使用层次遍历,将每一层的最后一个结点加入返回结果中。
4. 代码
public class Solution {
public List<Integer> rightSideView(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
if (root == null) {
return result;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
LinkedList<TreeNode> nextLevel = new LinkedList<>();
int lastVal = 0;
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
lastVal = node.val;
if (node.left != null) {
nextLevel.add(node.left);
}
if (node.right != null) {
nextLevel.add(node.right);
}
}
result.add(lastVal);
queue.addAll(nextLevel);
}
return result;
}
}