力扣102. 二叉树的层序遍历（O(n)时间复杂度，通过queue实现简单BFS）

最新推荐文章于 2024-08-23 01:05:03 发布

嘻·嘻

最新推荐文章于 2024-08-23 01:05:03 发布

阅读量520

点赞数 5

分类专栏：算法题解文章标签： leetcode 宽度优先算法数据结构

本文链接：https://blog.csdn.net/weixin_63800030/article/details/137211662

版权

算法题解专栏收录该内容

29 篇文章 1 订阅

订阅专栏

本文详细介绍了如何使用队列实现二叉树的层序遍历，通过递归和广度优先搜索(BFS)，记录节点层次并存储在vector中。同时分析了时间复杂度O(n)和空间复杂度O(n)。

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

Problem: 102. 二叉树的层序遍历

思路

记录层次，要么建立树时记录，要么遍历时记录（利用queue）

解题方法

queue<pair<TreeNode*,int>>q; 记录节点及其层次
vector<vector> res; res[i]表示第i层(从0开始)的所有节点

复杂度

时间复杂度:

$O (n)$

空间复杂度:

$O (n)$

Code

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<pair<TreeNode*,int>>q;// <节点，层数>
        q.push(make_pair(root,0));
        vector<vector<int>> res;// res[i]表示存储第i层的所有节点
        if(root==NULL)return res;
        res.push_back(vector<int>{root->val});
        int k=0;
        // bfs
        while(!q.empty()){
            TreeNode* tmp = q.front().first;
            k = q.front().second+1;// 下面会访问当前节点的两个孩子，所以层数加1
            q.pop();
            // 左孩子
            if(tmp->left!=NULL){
                q.push(make_pair(tmp->left,k));
                if(k>=res.size()){// 说明该层次还不存在，直接新建
                    vector<int> i;
                    res.push_back(i);
                }
                res[k].push_back(tmp->left->val);
            }
            // 右孩子
            if(tmp->right!=NULL){
                q.push(make_pair(tmp->right,k));
                if(k>=res.size()){
                    vector<int> i;
                    res.push_back(i);
                }
                res[k].push_back(tmp->right->val);
            }
        }
        return res;
    }
};