199. 二叉树的右视图

问题描述

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

bfs树的层次遍历

同其他的一些题，使用for循环区分该层的结点，并记录每层的最后一个结点为答案即可。

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> res;
        queue<TreeNode*> q;
        if (root != nullptr) {
            q.push(root);
        }
        while(!q.empty()) {
            int layer_size = q.size();
            for (int i = 0; i < layer_size; i++) {
                TreeNode* t = q.front();
                q.pop();
                if (t->left) {
                    q.push(t->left);
                } if (t->right) { // 这里不能else if 要考虑左右子节点同时存在的情况
                    q.push(t->right);
                }
                if (i == layer_size-1) {
                    res.push_back(t->val);
                }
            }
        }
        return res;
    }
};