力扣 hot100 Day46

199. 二叉树的右视图

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

//自己写的
class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> result;
        if (!root) return result;
    
        queue<TreeNode*> q;
        q.push(root);
    
        while (!q.empty()) {
            int levelSize = q.size();     
            for (int i = 0; i < levelSize; ++i) {
                TreeNode* node = q.front();
                q.pop();
                if(i==levelSize-1)
                {
                    result.push_back(node->val);
                }            
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }        
        }    
        return result;
    }
};

其实就是返回层序遍历中每行的最后一个值，在其基础上小改即可

重温一遍层序遍历逻辑吧

利用队列存放每层节点，遍历到每个节点时，弹出该节点，同时队尾加入该节点的左右节点

levelSize的设定比较巧妙，从最开始root节点逐渐推演，对应每层节点个数