利于队列求二叉树的右视图

leetcode 199. 二叉树的右视图

题目链接

给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:

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

思路

利用一个队列每次保存二叉树一层的节点，从右到左存，队首元素即右视图在在前层可见的节点

	struct TreeNode {
     	int val;
     	TreeNode *left;
     	TreeNode *right;
     	TreeNode() : val(0), left(nullptr), right(nullptr) {}
     	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
     	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x),left(left), right(right) {}
     };

    vector<int> rightSideView(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<int> result;
        while (!que.empty()) {
            int size = que.size();
            result.push_back(que.front()->val);		//队首元素添加到结果集
            for (int i = 0; i < size; i++) { 		//将队列中所有的节点的非空左右孩子进入队列，先进右孩子在进左孩子
                TreeNode* node = que.front();
                que.pop();
                if (node->right) que.push(node->right);
                if (node->left) que.push(node->left);
            }
        }
        return result;
    }