题目描述:
给定一个二叉树的 根节点 root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
思路:
1)结合二叉树的层序遍历,使用广度优先,只将每一层的最后一个节点输出即可!
2)DFS:使用深度优先也就是递归解决层序遍历,需要每次都将节点的 深度 作为参数传入
本题的节点深度,也就是depth++,要放对位置,记录好每一个节点正确的位置!放在if外面
代码:
1)广度优先
class Solution {
public List<Integer> rightSideView(TreeNode root) {
//依旧是层序遍历,但是此时每层只输出最后一个
List<Integer> res = new ArrayList<>();
if(root == null) return res;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
int size = queue.size();//记录这一层的长度,也就是节点数
//将最后一个节点之前的节点全部弹出
for(int i = 0;i < size;i++){
TreeNode temp = queue.poll();
//不在结果集中添加,只往队列中加节点
if(i == size - 1) res.add(temp.val);
if(temp.left != null) queue.offer(temp.left);
if(temp.right != null) queue.offer(temp.right);
}
}
return res;
}
}
2)DFS深度优先
class Solution {
public List<Integer> rightSideView(TreeNode root) {
//DFS
List<Integer> res = new ArrayList<>();
dfs(res,0,root);
return res;
}
//像这种的层序遍历,使用DFS实现,每次需要将深度 作为参数传入!
void dfs(List<Integer> list,int depth,TreeNode node){//1.先确定返回值、和传入的参数
if(node == null) return;//2.设置终止条件
//3.设置主体逻辑
//size从0开始,此时表示是第一次到达该层
if(depth == list.size()){
list.add(node.val);
}
//这个depth加1的位置怎么判断
depth++;
//按照从右往左进入
dfs(list,depth,node.right);
dfs(list,depth,node.left);
}
}