https://leetcode.cn/problems/binary-tree-right-side-view/description/?envType=study-plan-v2&envId=top-interview-150
其实就是让我们返回每一层最右边的节点的,首先想到的就是把每一层的节点从左往右排列出来然后添加最后一个节点,直到把所有层添加完。怎么获取各个层的节点呢,我们可以使用二叉树的层序遍历,使用队列来存储每一层的节点
class Solution {
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;
}
}
public List<Integer> rightSideView(TreeNode root) {
if(root == null) return new ArrayList<>();
if(root.left == null && root.right == null) return List.of(root.val);
Queue<TreeNode> queue1 = new LinkedList<>();//存储当前层的节点
Queue<TreeNode> queue2 = new LinkedList<>();//存储下一层的节点
queue1.add(root);
List<Integer> res = new ArrayList<>();
while (!queue1.isEmpty()) {
while(!queue1.isEmpty()){
if(queue1.size() == 1) { // 如果当前层只有一个节点,那么这个节点就是最右边的节点
res.add(queue1.peek().val);
}
TreeNode cur = queue1.poll();
// 填充下一层
if(cur.left != null) queue2.add(cur.left);
if(cur.right != null) queue2.add(cur.right);
}
//遍历完当前层后,交换queue1和queue2,遍历下一层
Queue<TreeNode> temp = queue1;
queue1 = queue2;
queue2 = temp;
}
return res;
}
}