


层序遍历的时候,判断是否遍历到单层的最后面的元素,如果是,就放进result数组中,随后返回result就可以了。
cpp
class Solution {
public:
vector<int> rightSideView(TreeNode* root) {
// 创建一个队列用于BFS层序遍历
queue<TreeNode*> que;
// 如果根节点不为空,将其加入队列
if (root != NULL) que.push(root);
// 存储结果的向量,将保存二叉树的右视图
vector<int> result;
// 当队列不为空时,继续遍历
while (!que.empty()) {
// 获取当前层的节点数量
int size = que.size();
// 遍历当前层的所有节点
for (int i = 0; i < size; i++) {
// 取出队首节点
TreeNode* node = que.front();
que.pop();
// 如果是当前层的最后一个节点(最右侧节点),将其值加入结果
// 因为队列是先进先出,所以最后一个节点就是该层最右边的节点
if (i == (size - 1)) result.push_back(node->val);
// 将当前节点的左子节点加入队列(如果存在)
if (node->left) que.push(node->left);
// 将当前节点的右子节点加入队列(如果存在)
if (node->right) que.push(node->right);
}
}
// 返回右视图结果
return result;
}
};