题目描述:
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
方法1:
主要思路:
(1)直观的想到,可以使用层次遍历,层次遍历每次从右到左执行,这样在遍历每层的时候,可以直接输出头结点;
(2)为了实现层次遍历,这里考虑使用队列辅助,既每次将下一层的结点按照先右结点,再左结点的顺序压入到队列中,这样队列在弹出的时候,就可以先弹出每层的右侧的第一个结点;
(3)每次弹出一层,弹出的数量是该层的结点的数量,在弹出的过程中,压入下一层的结点;
/**
* 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) {
if(root==NULL)
return vector<int>();
queue<TreeNode*> q;
//先压入根节点
q.push(root);
vector<int>res;//存储结果
//若当前队列非空
while(!q.empty()){
//先获得当前队列的大小,既当前层的结点的数量
int cur_size=q.size();
//将前队列的头元素,既当前层的右侧的第一个结点的值压入到结果中
res.push_back(q.front()->val);
TreeNode* tmp=NULL;
//弹出当前层,压入下一层
while(cur_size){
--cur_size;//保证只是弹出当前层
tmp=q.front();
q.pop();
if(tmp->right)//非空,则压入
q.push(tmp->right);
if(tmp->left)//非空,则压入
q.push(tmp->left);
}
}
return res;
}
};