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;
    }
};
相关推荐
Asmalin2 小时前
【代码随想录day 29】 力扣 135.分发糖果
算法·leetcode·职场和发展
微笑尅乐2 小时前
多解法详解与边界处理——力扣7.整数反转
算法·leetcode·职场和发展
夏鹏今天学习了吗2 小时前
【LeetCode热题100(31/100)】K 个一组翻转链表
算法·leetcode·链表
薰衣草23332 小时前
力扣——位运算
python·算法·leetcode
未知陨落2 小时前
LeetCode:83.打家劫舍
算法·leetcode
Pluchon2 小时前
硅基计划4.0 算法 字符串
java·数据结构·学习·算法
三年呀3 小时前
共识算法的深度探索:从原理到实践的全面指南
算法·区块链·共识算法·分布式系统·区块链技术·高性能优化
alex1003 小时前
BeaverTails数据集:大模型安全对齐的关键资源与实战应用
人工智能·算法·安全
麦格芬2303 小时前
LeetCode 416 分割等和子集
数据结构·算法
2401_841495644 小时前
【自然语言处理】Universal Transformer(UT)模型
人工智能·python·深度学习·算法·自然语言处理·transformer·ut