199. 二叉树的右视图
描述
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例
示例1
输入:[1,2,3,null,5,null,4]
输出:[1,3,4]
示例2
输入:[1,null,3]
输出:[1,3]
示例3
输入:root = []
输出:[]
链接
https://leetcode.cn/problems/binary-tree-right-side-view/
解题思路
思路一: 广度优先遍历
层序遍历的时候,判断是否遍历到单层的最后面的元素,如果是,就放进result数组中,随后返回result就可以了。
/**
* @param {TreeNode} root
* @return {number[]}
*/
var rightSideView = function(root) {
if (root === null) return [];
var res = [], queue = [root];
while(queue.length) {
var currentLevelSize = queue.length;
for (var i = 0 ;i < currentLevelSize; i++) {
var node = queue.shift();
// 将当前层的最后一个节点放入结果列表
(i == currentLevelSize - 1) && res.push(node.val);
if (node.left != null) queue.push(node.left);
if (node.right != null) queue.push(node.right);
}
}
return res;
};
时间复杂度: O(n), 其中 n 是二叉树的节点数
空间复杂度: O(n)
思路二: 深度优先搜索
思路: 我们按照 「根结点 -> 右子树 -> 左子树」 的顺序访问,就可以保证每层都是最先访问最右边的节点的。
(与先序遍历 「根结点 -> 左子树 -> 右子树」 正好相反,先序遍历每层最先访问的是最左边的节点)
/**
* @param {TreeNode} root
* @return {number[]}
*/
var rightSideView = function (root) {
if (root === null) return [];
var res = [];
var dfs = function (node, depth) {
if (node === null) return;
// 如果当前节点所在深度还没有出现在res里,说明在该深度下当前节点是第一个被访问的节点,因此将当前节点加入res中。
if (depth == res.length) {
res.push(node.val);
}
// 先访问 当前节点,再递归地访问 右子树 和 左子树
dfs(node.right, depth + 1);
dfs(node.left, depth + 1);
}
dfs(root, 0);
return res;
}
时间复杂度: O(n), 其中 n 是二叉树的节点数
空间复杂度: O(n)
参考资料
https://leetcode.cn/problems/binary-tree-right-side-view/solution/jian-dan-bfsdfs-bi-xu-miao-dong-by-sweetiee/