LeetCode刷题笔记 二叉树 层次遍历

​ 通常使用广度优先搜索进行层次遍历。注意,不需要使用两个队列来分别存储当前层的节点和下一层的节点,因为在开始遍历一层的节点时,当前队列中的节点数就是当前层的节点数,只要控制遍历这么多节点数,就能保证这次遍历的都是当前层的节点。

102 二叉树的层序遍历

实现二叉树的层序遍历

输入一个二叉树,输出一个二维数组,表示二叉树层序遍历的结果

输入:[3,9,20,null,null,15,7]

 3
/ \
9  20
 /  \
15   7

输出:

[
[3],
[9,20],
[15,7]
]

解析:

​ 通常使用广度优先搜索进行层次遍历,使用一个队列存储当前层的所有节点。

​ 在开始遍历一层的节点时,当前队列中的节点数就是当前层的节点数。

​ 只要控制遍历这么多节点数,每遍历一个当前层的节点,将其出队列同时将其子节点入队列。

​ 通过这种操作就能保证每次遍历的队列中都是当前层的节点。

class Solution {
    
    
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    
    
        vector<vector<int>> res;
        if(root==nullptr){
    
    
            return res;
        }
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()){
    
    
            int queLen = que.size();
            vector<int> levelElem;
            for(int i=0;i<queLen;++i){
    
    
                auto node = que.front();
                que.pop();
                levelElem.push_back(node->val);
                if(node->left){
    
    
                    que.push(node->left);
                }
                if(node->right){
    
    
                    que.push(node->right);
                }
            }
            res.push_back(levelElem);
        }
        return res;
    }
};

637 二叉树的层平均值

给定一个二叉树,求每一层的节点值的平均数。

输入是一个二叉树,输出是一个一维数组,表示每层节点值的平均数。

输入:

3
/ \
9  20
/  \
15   7

输出:[3, 14.5, 11]
解释:第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

解析:

​ 使用一个先入先出的队列对二叉树进行层次遍历,遍历每一层的过程中累计该层的总和并在最后将平均值加入结果集。

​ 使用双层循环完成每层均值计算:外循环逐层遍历二叉树;内循环遍历队列中当前层的所有节点,计算当前层节点值均值,并将下一层节点压入队列。

class Solution {
    
    
public:
    vector<double> averageOfLevels(TreeNode* root) {
    
    
        vector<double> ans;
        if(!root){
    
    
            return ans;
        }
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
    
    
            int len = q.size();
            double sum = 0;
            for(int i=0;i<len;++i){
    
    
                auto node = q.front();
                q.pop();
                sum+=node->val;
                if(node->left){
    
    
                    q.push(node->left);
                }
                if(node->right){
    
    
                    q.push(node->right);
                }
            }
            ans.push_back(sum/len);
        }
        return ans;
    }
};

513 找树左下角的值

给定一个二叉树,找出该二叉树的 最底层 最左边 节点的值

输入一个二叉树,输出一个整型值表示该二叉树左下角的值

输入: root = [2,1,3]
输出: 1

解析:

​ 本题很容易想到使用层次遍历解决,因为二叉树的 最底层 最左边 节点的值就是层次遍历最后一层的第一个节点值。

​ 所以,本题仅需要得到二叉树的层次遍历结果,然后将最后一层的第一个值返回即可。

class Solution {
    
    
public:
    int findBottomLeftValue(TreeNode* root) {
    
    
        int ans = 0;
        if(!root){
    
    
            return ans;
        }
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
    
    
            int queLen = q.size(); // 要记录最初的队列长度,不然在后续push和pop操作中会影响该值
            ans = q.front()->val; // 记录当前层的第一个值
            for(int i=0;i<queLen;++i){
    
    
                auto node = q.front();
                q.pop();
                if(node->left){
    
    
                    q.push(node->left);
                }
                if(node->right){
    
    
                    q.push(node->right);
                }
            }
        }
        return ans;
    }
};

参考资料

LeetCode 101:和你一起轻松刷题(C++) 第 14 章 指针三剑客之树

Guess you like

Origin blog.csdn.net/qq_41773806/article/details/121281764