Day23-二叉树的层序遍历(广度优先搜素)

今天只看了层序遍历

思路如下:一层层的去吧结果遍历到结果数组中。

层序遍历结果为:[1, 2, 3, 4, 5, 6]。

前面迭代方法是用栈去模拟的,那么层序遍历可以用队列去模拟:

先把根节点加入队列,然后在队列弹出来元素之前用一个node去指向它,弹出来元素之后先把node->val加入一维数组,然后用node去把它的左右孩子加入队列。

每遍历完一层的节点之后把一维数组加入结果数组。

个人感觉需要注意的是一维数组和node节点的创建一定是要在while循环里面。因为需要每次重新计算当前队列的大小!!!

这个可以当作一个模板:

层序遍历的思路基本上就是这样了。

cpp 复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;    //存放结果数组
        queue<TreeNode *> que;  //存储每一层的节点
        if(root == nullptr) return res; //根节点不为空的话就开始遍历
        que.push(root);
        while(!que.empty()){        //
            int n = que.size();     //每循环一次都要重新计算
            vector<int> ans;        //存放每一层的结果
            for(int i = 0;i<n;i++){     //开始遍历
                TreeNode *node = que.front();       //一定要创建一个节点去存储队列的元素
                que.pop();                          //和前序遍历的思路差不多
                ans.push_back(node->val);           //先把元素弹出然后左右孩子加入队列
                if(node->left) que.push(node->left);        //
                if(node->right) que.push(node->right);
            }
            res.push_back(ans);                 //每遍历完一层就把当前的数组加入结果
        }    
        return res;
    }
};
相关推荐
心中有国也有家2 小时前
cann-recipes-infer:昇腾 NPU 推理的“菜谱集合”
经验分享·笔记·学习·算法
绝知此事2 小时前
【算法突围 01】线性结构与哈希表:后端开发的收纳术
java·数据结构·算法·面试·jdk·散列表
碧海银沙音频科技研究院2 小时前
通话AEC与语音识别AEC的软硬回采链路
深度学习·算法·语音识别
csdn_aspnet3 小时前
Python 算法快闪 LeetCode 编号 70 - 爬楼梯
python·算法·leetcode·职场和发展
LuminousCPP4 小时前
数据结构 - 线性表第四篇:C 语言通讯录优化升级全记录(踩坑 + 思考)
c语言·开发语言·数据结构·经验分享·笔记·学习
AI算法沐枫4 小时前
深度学习python代码处理科研测序数据
数据结构·人工智能·python·深度学习·决策树·机器学习·线性回归
m0_629494736 小时前
LeetCode 热题 100-----26.环形链表 II
数据结构·算法·leetcode·链表
壹号用户6 小时前
用队列实现栈
数据结构·算法
做人求其滴6 小时前
面试经典 150 题 380 274
c++·算法·面试·职场和发展·力扣