[M二叉树] lc199. 二叉树的右视图(dfs+自顶向下+好题)

1. 题目来源

链接：199. 二叉树的右视图

题单：

• 11. 链表、二叉树与一般树（前后指针/快慢指针/DFS/BFS/直径/LCA）
  • §2.2 自顶向下 DFS
  • §2.13 BFS

2. 题目解析

思路：

• 换做是 bfs 应该非常好理解，只需要记录每一层的最后一个树节点即可。
• dfs 的话，需要注意下搜索顺序，因为是右视图，所以需要优先从右侧开始搜起。
• 记录一个答案数组。当树的高度和答案数组中的元素一致时，说明当前节点还没有被其他节点所挡住，记录该节点值到答案数组，继续向右儿子方向继续搜索，同时高度加 1。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> res;
    void dfs(TreeNode* root, int hi) {
        if (!root) return ;
        if (hi == res.size()) res.push_back(root->val);

        dfs(root->right, hi + 1);
        dfs(root->left, hi + 1);
        return ;
    }
    vector<int> rightSideView(TreeNode* root) {
        dfs(root, 0);
        return res;
    }
};