leetcode 199二叉树的右视图

可以，下面我把代码几乎每一行都加上注释，再给你总结一下。

复制代码

class Solution {
public:

    // ans 用来保存右视图结果
    // 每一层只保存一个节点
    // 例如 ans = [1, 3, 4]
    // 表示第0层记录1，第1层记录3，第2层记录4
    vector<int> ans;

    // dfs：深度优先搜索
    // root 表示当前遍历到的节点
    // depth 表示当前节点所在的层数
    // 根节点从第0层开始
    void dfs(TreeNode* root, int depth) {

        // 如果当前节点为空，说明走到底了
        // 直接返回，结束当前这次递归
        if (root == nullptr) {
            return;
        }

        // 如果当前层数 depth
        // 刚好等于 ans 中已经记录的层数
        // 说明这一层还没有记录任何节点
        if (depth == ans.size()) {

            // 因为我们是先遍历右子树
            // 所以这一层第一次遇到的节点
            // 就是这一层最右边能看到的节点
            ans.push_back(root->val);
        }

        // 先递归遍历当前节点的右子树
        // depth + 1 表示进入下一层
        dfs(root->right, depth + 1);

        // 再递归遍历当前节点的左子树
        // 同样进入下一层
        dfs(root->left, depth + 1);
    }

    // 主函数：返回二叉树的右视图
    vector<int> rightSideView(TreeNode* root) {

        // 从根节点开始递归
        // 根节点所在层数为0
        dfs(root, 0);

        // 返回最终记录的每一层最右边节点
        return ans;
    }
};

比如这棵树：

复制代码

遍历顺序因为是：

复制代码

根 -> 右 -> 左

所以大致会先走：

复制代码

1
→ 3
→ 4
→ 再回来走2
→ 5

一开始：

复制代码

ans = []

访问 1：

复制代码

depth = 0
ans.size() = 0

相等，所以记录：

复制代码

ans = [1]

访问 3：

复制代码

depth = 1
ans.size() = 1

相等，所以：

复制代码

ans = [1, 3]

访问 4：

复制代码

depth = 2
ans.size() = 2

相等，所以：

复制代码

ans = [1, 3, 4]

之后再访问 2：

复制代码

depth = 1
ans.size() = 3

不相等，说明第1层已经记录过了，所以不加入。

再访问 5：

复制代码

depth = 2
ans.size() = 3

也不相等，所以不加入。

最终：

复制代码

[1, 3, 4]

总结

这道题的核心有三个点：

第一，depth 表示当前节点在哪一层：

复制代码

根节点 depth = 0
下一层 depth = 1
再下一层 depth = 2

第二，ans.size() 表示目前已经记录了多少层 。因为 ans 每一层只存一个节点：

复制代码

ans = []          → 记录0层
ans = [1]         → 记录1层
ans = [1,3]       → 记录2层
ans = [1,3,4]     → 记录3层

所以：

复制代码

if (depth == ans.size())

表示：

当前来到了一个还没有记录过的新层。

第三，一定要：

复制代码

dfs(root->right, depth + 1);
dfs(root->left, depth + 1);

先右后左。

这样每一层第一次被访问到的节点，才是右边看到的节点。

你最后可以把整题记成一句话：

DFS 按"根→右→左"遍历，用 depth == ans.size() 判断是否第一次来到这一层，第一次来到就记录当前节点。