199. 二叉树的右视图 C++实现

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

示例:

输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
//右视图是按照层打印的最后一个节点
//首先实现按照层打印二叉树
//存每一层最后一个节点
    vector<int> rightSideView(TreeNode* root) {
        //使用队列
        vector<int> data;
        queue<TreeNode *> que;
        if(root == NULL) return data;
        que.push(root);
        int current = 1; // 用于记录当前层有几个节点
        int next = 0; //用与记录下一层有几个节点；
        while(!que.empty()) {
            TreeNode* tmp =  que.front();
            que.pop();
            if(tmp->left != NULL){
                que.push(tmp->left);
                next++;
            }
            if(tmp->right != NULL){
                que.push(tmp->right);
                next++;
            }
            current--;
            if(current == 0) {
                data.push_back(tmp->val);
                current = next;
                next = 0;
            }
        }
        return data;
    }
};