题目描述:
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
本题核心思想就是遍历二叉树,取得每一层最右边的第一个节点值。
关键点在于有几种遍历方法呢?
第一种:
广度优先搜索(BFS遍历方法)
class Solution {
public List<Integer> rightSideView(TreeNode root) {
//res去储存最后输出的结果
List<Integer> res = new LinkedList<>();
if(root == null){
return res;
}
//Queue去储存每一层的节点
Queue<TreeNode> q= new LinkedList<>();
//Queue首先加入根节点
q.offer(root);
//从上往下遍历
while(!q.isEmpty()){
int sz = q.size();
//先记录根节点,第一层也就是根节点所在的层,遍历完后加入res中
TreeNode last= q.peek();
//遍历每一层
for(int i =0; i < sz; i++){
//这一步, 可以理解为先取出根节点
TreeNode cur = q.poll();
// 从右边往左遍历,如果当前的节点有右子树,就把右子树的节点放入Queue中
if(cur.right != null){
q.offer(cur.right);
}
if(cur.left != null){
q.offer(cur.left);
}
}
res.add(last.val);
}
return res;
}
}
第二种:
深度优先搜索(DFS遍历方法)
class Solution {
List<Integer> res = new LinkedList<>();
//记录每一层的深度
int depth = 0;
public List<Integer> rightSideView_DFS(TreeNode root) {
traverse(root);
return res;
}
// 二叉树遍历函数,从上往下遍历
void traverse(TreeNode root) {
if (root == null) {
return;
}
//每一次递归进入遍历函数,depth加一
depth++;
//basecase,由于先遍历右子树,一开始进入遍历函数res的size是0,而depth此时是1,所以根节点的右子树节点会陆续加入res中,直到其相等跳出递归。
if (res.size() < depth) {
// 这一层还没有记录值
// 说明 root 就是右侧视图的第一个节点
res.add(root.val);
}
// 先遍历右子树再遍历左子树,先遍历的就是右侧节点
traverse(root.right);
traverse(root.left);
//从上到下遍历完,更新depth值
depth--;
}