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;
    }
};
相关推荐
你撅嘴真丑1 天前
第九章-数字三角形
算法
uesowys1 天前
Apache Spark算法开发指导-One-vs-Rest classifier
人工智能·算法·spark
ValhallaCoder1 天前
hot100-二叉树I
数据结构·python·算法·二叉树
董董灿是个攻城狮1 天前
AI 视觉连载1:像素
算法
智驱力人工智能1 天前
小区高空抛物AI实时预警方案 筑牢社区头顶安全的实践 高空抛物检测 高空抛物监控安装教程 高空抛物误报率优化方案 高空抛物监控案例分享
人工智能·深度学习·opencv·算法·安全·yolo·边缘计算
孞㐑¥1 天前
算法——BFS
开发语言·c++·经验分享·笔记·算法
月挽清风1 天前
代码随想录第十五天
数据结构·算法·leetcode
XX風1 天前
8.1 PFH&&FPFH
图像处理·算法
NEXT061 天前
前端算法:从 O(n²) 到 O(n),列表转树的极致优化
前端·数据结构·算法
代码游侠1 天前
学习笔记——设备树基础
linux·运维·开发语言·单片机·算法