给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4] 输出: [1, 3, 4] 解释: 1 <--- / \ 2 3 <--- \ \ 5 4 <---
题目分析:本题的大意即求一棵二叉树每层的最后一个结点值。利用BFS按层进行访问,保存每层的最后一个结点值即可。
代码展示:
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
vector<int> ans;
if(root==NULL)
return ans;
queue<TreeNode*> q;
q.push(root);
ans.push_back(root->val);
while(!q.empty()){
int count = q.size();
vector<int> vec;
for(int i=1;i<=count;i++){
TreeNode* temp = q.front();
q.pop();
if(temp->left!=NULL){
q.push(temp->left);
vec.push_back(temp->left->val);
}
if(temp->right!=NULL){
q.push(temp->right);
vec.push_back(temp->right->val);
}
}
if(vec.size())
ans.push_back(vec[vec.size()-1]);
}
return ans;
}
};