力扣 102 二叉树的层序遍历【广度优先遍历】

力扣 102 二叉树的层序遍历【广度优先遍历】

全部刷题与学习记录

【C++刷题学习笔记目录】

【C++百万并发网络通信-笔记目录】

原题目

题目地址:102. 二叉树的层序遍历

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:
二叉树:[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层序遍历结果:

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

注:本题与 剑指 32_2 分行从上到下打印二叉树【广度优先遍历】【层序遍历】 相同,属于二刷,但是做起来还是没有那么顺手,因为对与队列的处理不太娴熟


考查知识点

层序遍历、广度优先遍历


自己的第一遍解法

层序遍历的代码是比深度优先难想一点的,因为用到了一个队列,并且需要按行区分数组

要点就是在访问队列当前队首节点时,要将访问节点的子节点(如果有的话)加入队列,放在队尾。因为要按行区分数组,所以要有一个计数的概念,要在一行所有节点都加入到一个数组中以后,自动开始下一数组。主要是这个计数器的实现,其实用两层循环就好了

class Solution {
    
    
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    
    
        if (root == nullptr)    return {
    
    };
        queue<TreeNode*> que;   //逐步存储某一层全部节点
        que.push(root);

        vector<vector<int>> res;
        while (!que.empty()) {
    
    
            vector<int> level;  //当前层所有数字
            int size = que.size();
            //将队列中(当前层)所有数字添加到数组中,自动形成分数组存储不同层
            for (int i = 0; i < size; ++i) {
    
    
                TreeNode* cur = que.front();
                que.pop();
                level.push_back(cur->val);
                if (cur->left) que.push(cur->left);
                if (cur->right) que.push(cur->right);
            }
            res.push_back(level);
        }
        return res;
    }
};

参考:

https://mp.weixin.qq.com/s/Gb3BjakIKGNpup2jYtTzog

猜你喜欢

转载自blog.csdn.net/weixin_44484715/article/details/114949811
今日推荐