一、题目描述
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:
输入: [1,null,3]
输出: [1,3]
示例 3:
输入: []
输出: []
二、解题
BFS
这题使用广度优先遍历,层次遍历,遍历每一层,将最后一个数据保存就是需要的返回集合。
/**
* Definition for a binary tree node.
* 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;
* }
* }
*/
class Solution {
public List<Integer> rightSideView(TreeNode root) {
//层次遍历,每层入队列,取队列的最后一个值即可。
//BFS
Deque<TreeNode> queue = new LinkedList<>();
List<Integer> res = new ArrayList<>();
if(root == null){
return res;
}
queue.add(root);
while(!queue.isEmpty()){
int size = queue.size();
for(int i = 0;i<size;i++){
TreeNode node = queue.poll();
if(i == size-1){
res.add(node.val);
}
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
}
}
return res;
}
}
DFS
DFS没有BFS那样好理解,按照根节点-右子树-左子树的顺序访问,保证每层访问的第一个节点是最右边的节点即可。
class Solution {
public List<Integer> rightSideView(TreeNode root) {
//DFS
List<Integer> res = new ArrayList<>();
if(root == null){
return res;
}
dfs(root,0,res);
return res;
}
public void dfs(TreeNode root,int depth,List<Integer> res){
if(root == null){
return;
}
if(depth == res.size()){
res.add(root.val);
}
depth++;
dfs(root.right,depth,res);
dfs(root.left,depth,res);
}
}