LeetCode:199. 二叉树的右视图(C++带详细注释)

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;//返回结果

    }
};

你可能感兴趣的:(Leetcode题目,二叉树,队列,数据结构,leetcode,算法)