【层序遍历专场练习Ⅰ】102. 二叉树的层序遍历、107. 二叉树的层次遍历Ⅱ、199. 二叉树的右视图、637. 二叉树的平均值

102. 二叉树的层序遍历

题目描述:给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)

解答:

遍历很简单,出队进行访问,访问结束后,左右节点分别入队。

但是有个问题,每一层访问多少个节点?

其实每次队中的元素即为这一层的所有元素,每次获取一下队列长度再进行遍历即可。此处容易写出BUG的点在于:使用固定大小size,不要使用que.size(),因为que.size是不断变化的

代码实现:

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size,不要使用que.size(),因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        return result;
    }
};

107. 二叉树的层次遍历Ⅱ

题目描述:给你二叉树的根节点 root ,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

解答:先按照上题进行层序遍历,先不写入结果中,记录每一层的节点数量,将所有节点的值存在一个数组中。然后按照每层的节点数从后往前遍历存储节点值的数组,反转后放入resutl。

代码实现:

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<int>layer, num;
        vector<vector<int>>result;
        queue<TreeNode*>que;
        if(root != NULL)
            que.push(root);
        while(!que.empty()){
            int size = que.size();
            num.push_back(size);//记录每一层的节点数量
            for (int i=0; i<size; i++){
                TreeNode* p = que.front();
                layer.push_back(p->val);//用一个数组存储按序所有节点值
                que.pop();
                if (p->left) que.push(p->left);
                if (p->right) que.push(p->right); 
            }
        }
        //从后往前遍历,将每一层的遍历结果存入result
        for (int i=num.size()-1, j=layer.size()-1; i>=0 ;i--){
            int len = num[i];
            vector<int>st;
            while(len){
                st.push_back(layer[j--]);
                len--;
            }
            reverse(st.begin(), st.end());//因为从左向右遍历,翻转
            result.push_back(st);
        }
        return result;
    }
};

199. 二叉树的右视图

题目描述:给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

解答:

        首先这道题目提到每一层的最右侧,想到应该采用层次遍历。

        其次考虑什么时候是那一侧的最右侧?

        在进行层次遍历时每一层的数量是知道的,在访问至这一层的最后一个元素时,其即为最右侧的元素(因为是从左至右遍历)。所以只需要修改层次遍历代码,在每一层访问至最后一个元素时存储到结果中即可。

代码实现:

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int>result;
        queue<TreeNode*>que;
        if (root != NULL)
            que.push(root);
        while(!que.empty()){
            int size = que.size();
            for (int i = 0; i<size; i++){
                TreeNode* p = que.front();
                que.pop();
                if (i == size-1)    result.push_back(p->val);//只存储最右侧元素即可
                if (p->left)    que.push(p->left);
                if (p->right)   que.push(p->right);
            }
        }
        return result;
    }
};

637. 二叉树的平均值

题目描述:给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

解答:思路很简单,层次遍历,遍历完一层后计算均值即可。

代码实现:

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double>result;
        queue<TreeNode*>que;
        if (root != NULL)
            que.push(root);
        while(!que.empty()){
            int size = que.size();
            double num = 0;
            for (int i = 0; i<size; i++){
                TreeNode* p = que.front();
                que.pop();
                num += p->val;
                if (i == size-1)    result.push_back(num/size);//一层遍历完后计算均值
                if (p->left)    que.push(p->left);
                if (p->right)   que.push(p->right);
            }
        }
        return result;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值