刷题day28：二叉树的层序遍历

最新推荐文章于 2024-09-12 19:02:20 发布

Kbbl_dh

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量59

点赞数

文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/Kbbl_dh/article/details/129300721

版权

使用队列进行二叉树的层次遍历，从根节点开始，逐层访问每个节点，先入队的节点先出队，依次添加左右子节点。代码中，创建一个队列存放节点，当队列非空时，取出当前层的所有节点并存储其值，然后将下一层的节点入队，最后返回结果。

摘要由CSDN通过智能技术生成

题意描述：

给你二叉树的根节点 root ，返回其节点值的层序遍历。（即逐层地，从左到右访问所有节点）。

需要借用一个辅助数据结构即队列来实现，队列先进先出，符合一层一层遍历的逻辑，而用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。

流程图如下：

完整C++代码如下：

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        vector<vector<int>> result;
        if(root != NULL){
            que.push(root);
        }
        while(!que.empty()){
            int size = que.size();
            vector<int> vec;
            for(int i = 0; i < size; i++){
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if(node->left){
                    que.push(node->left);
                }
                if(node->right){
                    que.push(node->right);
                }
            }
            result.push_back(vec);
        }
        return result;
    }
};

参考：代码随想录 (programmercarl.com)