二叉树层次遍历,记录层次/每层节点数/每层最左边最右边的节点

题目来源:leetcode
199. 二叉树的右视图

层次遍历用队列,但是如何记录层次是一个需要思考的地方。一种方法是提供两个变量记录本层节点数和下层节点数,还有一种很巧妙的方法,记录队列中元素的数量,就是当前层次节点数。

题解:
这道题思路是使用层次遍历,每层最右边一个节点就是右视图的一部分。如果求左视图,就取每层最左边的节点。

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        
        if(root==nullptr) return vector<int>{};

        vector<int>ans;//记录答案
        queue<TreeNode*>q;//队列用于层次遍历
        q.push(root);

        while(!q.empty())
        {
            int size = q.size();//当前层节点的个数!
            ans.push_back(q.front()->val);//看下面的进队顺序,队列最前面的就是本层最右的元素

            while(size--)//遍历当前层的元素!
            {
                TreeNode* temp=q.front();
                q.pop();
                if(temp->right) q.push(temp->right);//进队语句交换,可以实现 左视图
                if(temp->left) q.push(temp->left);
            }
        }
        return ans;
    }     
};

参考:@淹死在水里的鱼 https://leetcode-cn.com/problems/binary-tree-right-side-view/comments/

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值