1. 题目来源
链接:199. 二叉树的右视图
题单:
-
- 链表、二叉树与一般树(前后指针/快慢指针/DFS/BFS/直径/LCA)
- §2.2 自顶向下 DFS
- §2.13 BFS
2. 题目解析
思路:
- 换做是 bfs 应该非常好理解,只需要记录每一层的最后一个树节点即可。
- dfs 的话,需要注意下搜索顺序,因为是右视图,所以需要优先从右侧开始搜起。
- 记录一个答案数组。当树的高度和答案数组中的元素一致时,说明当前节点还没有被其他节点所挡住,记录该节点值到答案数组,继续向右儿子方向继续搜索,同时高度加 1。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) 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;
}
};