LeetCode:199. 二叉树的右视图(C++带详细注释)
/**
* 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) {//层序遍历二叉树,记录每一层最后一个元素
queue<TreeNode*> que;//定义层序遍历队列
vector<int> res; //结果数组
if(root==NULL) return res;//树空返回
que.push(root);//非空将根节点入队
while(!que.empty())//队列非空时循环
{
int len=que.size();//队列长度即树每一层节点个数
for(int i=0;i<len;i++)//依次遍历此层每个节点,结束后此层节点全出队,剩下一层节点
{
TreeNode*tep=que.front();//得队头元素
que.pop();//队头出队
if(i==len-1) res.push_back(tep->val);//记录最后一个元素
if(tep->left) que.push(tep->left);//左子树非空入队
if(tep->right) que.push(tep->right);//右子树非空入队
}
}
return res;//返回结果
}
};