代码随想录|二叉树的层序遍历

最新推荐文章于 2024-06-21 15:51:08 发布

weixin_51707660

最新推荐文章于 2024-06-21 15:51:08 发布

阅读量414

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_51707660/article/details/131906577

版权

本文主要介绍了二叉树的各种层序遍历问题，包括常规的层序遍历、自底向上的层序遍历、二叉树的右视图、层平均值计算、N叉树的层序遍历以及寻找每层的最大值。通过层序遍历的思路解决了一系列相关问题，并提供了相应的代码实现。

摘要由CSDN通过智能技术生成

代码随想录|二叉树的层序遍历

102.二叉树的层序遍历

【题目描述】
给你二叉树的根节点 root ，返回其节点值的层序遍历。（即逐层地，从左到右访问所有节点）。

【思路】
（1）使用队列进行每层节点的遍历
（2）使用size变量记录所在层还剩多少节点
（3）将每层的非空左右孩子插入队列中

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
   
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
   
        vector<vector<int>> result;
        queue<TreeNode*> que;
        if(root != nullptr){
   
            que.push(root);
        }
        while(!que.empty()){
   
            int size = que.size();
            vector<int> temp;
            while(size--){
   
                TreeNode* node = que.front();
                temp.push_back(node->val);
                que.pop();
                if(node->left){
   
                    que.push(node->left);
                }
                if(node->right){
   
                    que.push(node->right);
                }
            }
            result.push_back(temp);
        }
        return result;
    }
};

时间复杂度：O(n),每个节点进队出队各一次
空间复杂度：O(n),队列中元素的个数不超过 n个

107.二叉树的层次遍历II

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

【思路】
（1）将102题目的结果翻转就好了

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
   
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
   
        vector<vector<int>> result;
        queue<TreeNode*> que;
        if(root != nullptr){
   
            que.push(root);
        }
        while(!que.empty()){
   
            int size = que.size();
            vector<int> temp;
            while(size--){
   
                TreeNode* node = que.front();
                temp.push_back(node->val);
                que.pop();
                if(node->left){
   
                    que.push(node->left);
                }
                if(node->right){
   
                    que.push(node->right);
                }
            }
            result.push_back(temp);
        }
        reverse(result.begin(),result.end(<