199. 二叉树的右视图
题目:
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <—
/
2 3 <—
\
5 4 <—
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:二叉树的层遍历
想象自己站在它的右侧,我们看到的是每一层的最右的一个节点,所以我们按层遍历,保存每层最后一个节点就能得到正确结果。
class Solution {
public List<Integer> rightSideView(TreeNode root) {
LinkedList<TreeNode> que = new LinkedList<>();
List<Integer> ans = new ArrayList<>();
if(root == null) return ans;
que.add(root);
while(!que.isEmpty()) {
int s = que.size();
ans.add(que.getLast().val);
for(int i = 0; i < s; i++) {
TreeNode cur = que.pop();
if(cur.left != null) {
que.add(cur.left);
}
if(cur.right != null){
que.add(cur.right);
}
}
}
return ans;
}
}