给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:
输入: [1,null,3]
输出: [1,3]
示例 3:
输入: []
输出: []
提示:
- 二叉树的节点个数的范围是 [0,100]
- -100 <= Node.val <= 100
解题思路
1.二叉树的右视图 ==> 层序遍历中的每一层的最后一个节点
2.因此可以用层序遍历,然后遍历过程中只取每层的最右边一个即可
代码
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number[]}
*/
var rightSideView = function(root) {
if (!root) return [];
// 根节点入队
const q = [root];
// 返回值
const res = [];
while(q.length) {
// 当前层的数量
let len = q.length;
while(len--) {
const n = q.shift();
// 只放入最右边的数
if (!len) res.push(n.val);
n.left && q.push(n.left);
n.right && q.push(n.right);
}
}
return res;
};