199. 二叉树的右视图（JS实现）

最新推荐文章于 2022-02-14 19:36:24 发布

原创最新推荐文章于 2022-02-14 19:36:24 发布 · 584 阅读

1 ·

CC 4.0 BY-SA版权

本文为博主原创文章，转载请务必附上文章网址，谢谢。

剑指offer 专栏收录该内容

153 篇文章

订阅专栏

本文介绍了一种解决二叉树右视图问题的算法，通过深度优先搜索遍历树的右分支，然后回溯到左分支，记录每个深度可见的节点值。适用于LeetCode上的一道经典算法题。

1 题目

给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <—
/
2 3 <—
\
5 4 <—

链接：https://leetcode-cn.com/problems/binary-tree-right-side-view

2 思路

这道题我们可以从对树进行中序遍历，但得先遍历树的右分支，再左分支，在遍历的过程中，记录每个节点的当前深度和当前遍历过程的最大深度，若当前节点的深度大于当前最大深度，说明该节点从右侧可以看见，则将其加入结果中

3代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var rightSideView = function(root) {
    let nums = [];
    if (!root) return nums;

    let stack = [];
    let p = root;
    let maxDepth = 0;
    let currentDepth = 0;
    while(p || stack.length > 0) {
        while(p) {      //遍历节点的右分支
            currentDepth++;
            if (currentDepth > maxDepth) {    //推入节点
                maxDepth++;
                nums.push(p.val);
            }
            stack.push([p, currentDepth]);
            p = p.right;
        }

        let node = stack.pop();    //回溯
        p = node[0].left;     //对节点的左分支进行遍历
        currentDepth = node[1];  //当前深度也要回溯
    }

    return nums;
};