[LeetCode][102]二叉树的层序遍历——遍历结果中每一层明显区分

题目

102. 二叉树的层序遍历

给定二叉树的根节点 root,返回节点值的层序遍历结果。即逐层地,从左到右访问所有节点。

示例 1:

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

示例 2:

输入:root = [1] 输出:[[1]]

示例 3:

输入:root = [] 输出:[]

提示:

树中节点数目在范围 [0, 2000] 内

-1000 <= Node.val <= 1000

思考

  • 二叉树的层序遍历是比较常见的题目
  • 此题的难点是如何把每层的节点归类为同一行,如果全部节点都放入队列中直接进行遍历,那么访问的节点可能超出一层
  • 是否需要一个变量,记录本层的节点个数。开始循环遍历队列,本次遍历根据本层的节点个数,将本层节点循环遍历记录下来,在这个过程中同时将本层节点可以访问到的下一层节点存入队列,由于先记录了本层的节点个数,所以这个循环不会超出本层
  • 在这种情况下,每次循环都将本层的节点全部输出,然后将下一层的节点全部加入,所以在开始输出和加入之前,队列中元素的个数就是本层所有节点的个数,所以可以像这样优雅地记录本层的节点个数
cpp 复制代码
for(int i=q.size(); i>0; --i)

解法1:简洁解法

cpp 复制代码
/**
 * 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>> ans;
        if(!root) return ans;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            vector<int> v;
            for(int i=q.size(); i>0; --i){
                v.push_back(q.front()->val);
                if(q.front()->left) q.push(q.front()->left);
                if(q.front()->right) q.push(q.front()->right);
                q.pop();
            }
            ans.push_back(v);
        }
        return ans;
    }
};

解法2:由二维数组联想到使用二维队列

  • 二维队列中每个小队列包含一层的所有元素
cpp 复制代码
/**
 * 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>> ans;
        if(!root) return ans;
        queue<queue<TreeNode*>> layerSeq;
        queue<TreeNode*> q;
        q.push(root);
        layerSeq.push(q);
        while(!layerSeq.empty()){
            vector<int> v;
            queue<TreeNode*> q;
            while(!layerSeq.front().empty()){
                v.push_back(layerSeq.front().front()->val);
                if(layerSeq.front().front()->left) q.push(layerSeq.front().front()->left);
                if(layerSeq.front().front()->right) q.push(layerSeq.front().front()->right);
                layerSeq.front().pop();
            }
            if(!q.empty()) layerSeq.push(q);
            layerSeq.pop();
            ans.push_back(v);
        }
        return ans;
    }
};
相关推荐
Beau_Will4 分钟前
数据结构-树状数组专题(1)
数据结构·c++·算法
迷迭所归处8 分钟前
动态规划 —— 子数组系列-单词拆分
算法·动态规划
爱吃烤鸡翅的酸菜鱼9 分钟前
Java算法OJ(8)随机选择算法
java·数据结构·算法·排序算法
寻找码源1 小时前
【头歌实训:利用kmp算法求子串在主串中不重叠出现的次数】
c语言·数据结构·算法·字符串·kmp
Matlab精灵1 小时前
Matlab科研绘图:自定义内置多款配色函数
算法·matlab
诚丞成1 小时前
滑动窗口篇——如行云流水般的高效解法与智能之道(1)
算法
带多刺的玫瑰2 小时前
Leecode刷题C语言之统计不是特殊数字的数字数量
java·c语言·算法
爱敲代码的憨仔3 小时前
《线性代数的本质》
线性代数·算法·决策树
yigan_Eins3 小时前
【数论】莫比乌斯函数及其反演
c++·经验分享·算法
阿史大杯茶3 小时前
AtCoder Beginner Contest 381(ABCDEF 题)视频讲解
数据结构·c++·算法