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;
    }
};
相关推荐
AI软著研究员7 小时前
程序员必看:软著不是“面子工程”,是代码的“法律保险”
算法
FunnySaltyFish7 小时前
什么?Compose 把 GapBuffer 换成了 LinkBuffer?
算法·kotlin·android jetpack
颜酱8 小时前
理解二叉树最近公共祖先(LCA):从基础到变种解析
javascript·后端·算法
地平线开发者1 天前
SparseDrive 模型导出与性能优化实战
算法·自动驾驶
董董灿是个攻城狮1 天前
大模型连载2:初步认识 tokenizer 的过程
算法
地平线开发者1 天前
地平线 VP 接口工程实践(一):hbVPRoiResize 接口功能、使用约束与典型问题总结
算法·自动驾驶
罗西的思考1 天前
AI Agent框架探秘:拆解 OpenHands(10)--- Runtime
人工智能·算法·机器学习
HXhlx1 天前
CART决策树基本原理
算法·机器学习
Wect1 天前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript
颜酱1 天前
单调队列:滑动窗口极值问题的最优解(通用模板版)
javascript·后端·算法