解题思路 :
优先遍历右子树节点,图中红色的节点是最右侧的节点,当后边遍历到绿色节点时,因为其深度和已遍历的右侧节点个数相同,所以这里不用加入结果中,只有其深度大于或者等于已遍历的节点的个数时,才能加入到结果中。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> res;
dfs(root, 0, res);
return res;
}
void dfs(TreeNode *node, int level, vector<int>& res) {
if (!node)
return;
if (level >= res.size()) {
res.push_back(node->val);
}
dfs(node->right, level + 1, res); //优先遍历右子树,只有当节点的层数大于树的高度时,才保存到res中去
dfs(node->left, level + 1, res);
}
};