给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:
输入: [1,null,3]
输出: [1,3]
示例 3:
输入: []
输出: []
提示:
二叉树的节点个数的范围是 [0,100]
思路:
- 通过 dfs 递归(反向前序遍历, 根、右、左节点)
- 递归时,传一个 d(层数)作为参数
- 当前层如果是第一次递归到,则将当前节点加入结果数组,否则,继续递归
- 最后将结果集返回
/**
* 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:
void dfs(TreeNode* root, vector<int> &res, int d){
if(root == nullptr) return;
if(res.size() < d){
res.push_back(root -> val);
}
dfs(root -> right, res, d + 1);
dfs(root -> left, res, d + 1);
}
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
dfs(root, res, 1);
return res;
}
};