[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;
    }
};
相关推荐
JSU_曾是此间年少5 分钟前
数据结构——线性表与链表
数据结构·c++·算法
sjsjs1111 分钟前
【数据结构-合法括号字符串】【hard】【拼多多面试题】力扣32. 最长有效括号
数据结构·leetcode
此生只爱蛋1 小时前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
咕咕吖2 小时前
对称二叉树(力扣101)
算法·leetcode·职场和发展
九圣残炎2 小时前
【从零开始的LeetCode-算法】1456. 定长子串中元音的最大数目
java·算法·leetcode
lulu_gh_yu2 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法
丫头,冲鸭!!!2 小时前
B树(B-Tree)和B+树(B+ Tree)
笔记·算法
Re.不晚3 小时前
Java入门15——抽象类
java·开发语言·学习·算法·intellij-idea
为什么这亚子3 小时前
九、Go语言快速入门之map
运维·开发语言·后端·算法·云原生·golang·云计算
4 小时前
开源竞争-数据驱动成长-11/05-大专生的思考
人工智能·笔记·学习·算法·机器学习