力扣:199.二叉树的右视图
代码随想录解析
题目:
给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
思路:
按层序遍历,当要出队列的元素是每一行的最后一个元素时,元素值入 vector 中。
代码:
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
queue<TreeNode*>que;
if(root!=nullptr) que.push(root);
vector<int> v;
while(!que.empty()){
int size = que.size();
for(int i = 0; i < size; i++){
TreeNode*node = que.front();
que.pop();
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
if(i == size-1) v.push_back(node->val);
}
}
return v;
}
};