题目:
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [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
思路:层序遍历,取每一层的最后一个元素。可以用两个list分别存放上一层与下一层
Java代码如下:
public List<Integer> rightSideView(TreeNode root) {
List<Integer> result = new LinkedList<>();
List<TreeNode> list1 = new LinkedList<>(); //用于存放上一层的节点
List<TreeNode> list2 = new LinkedList<>(); //用于存放下一层的节点
if(root == null){
return result;
}
list1.add(root);
while (!list1.isEmpty()){
result.add(list1.get(list1.size()-1).val);
for(TreeNode treeNode:list1){
if(treeNode.left!=null){
list2.add(treeNode.left);
}
if(treeNode.right!=null){
list2.add(treeNode.right);
}
}
list1.clear();
list1.addAll(list2);
list2.clear();
}
return result;
}