199. 二叉树的右视图 (BFS和DFS)
问题:给定一个二叉树的 根节点 root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
输入: [1,null,3]
输出: [1,3]
输入: []
输出: []
思路:
- BFS 层序遍历,记录每一层最右节点即可,两种方式 正序记录最后一个,逆序记录每一层第一个;
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if(root == null){
return ans;
}
Deque<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
int size = queue.size();
for(int i = 0; i < size; i++){
TreeNode cur = queue.poll();
if(i == size - 1){
ans.add(cur.val);
}
if(cur.left != null){
queue.offer(cur.left);
}
if(cur.right != null){
queue.offer(cur.right);
}
}
}
return ans;
}
}
- DFS
递归过程中,记录每一层的最右节点,先遍历右子树,再遍历左子树,这样可以保证先访问这一层的最右节点。若当前层已经记录了最右节点,则不记录,若未记录,则加入集合。
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if(root == null){
return ans;
}
dfs(root, 0, ans);
return ans;
}
public void dfs(TreeNode node, int level, List<Integer> ans){
if(node == null){
return;
}
//当前层未记录最右节点,当前节点就是最右节点
if(level == ans.size()){
ans.add(node.val);
}
dfs(node.right, level + 1, ans);
dfs(node.left, level + 1, ans);
}
}
整理思路,记录博客,以便复习。若有误,望指正~