前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >二叉树的层序遍历

二叉树的层序遍历

作者头像
狼啸风云
发布2024-01-28 09:44:24
1370
发布2024-01-28 09:44:24
举报

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

示例 1:

代码语言:javascript
复制
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]

示例 2:

代码语言:javascript
复制
输入:root = [1]
输出:[[1]]

示例 3:

代码语言:javascript
复制
输入:root = []
输出:[]

提示:

  • 树中节点数目在范围 [0, 2000] 内
  • -1000 <= Node.val <= 1000

思路和算法

我们可以用广度优先搜索解决这个问题。

我们可以想到最朴素的方法是用一个二元组 (node, level) 来表示状态,它表示某个节点和它所在的层数,每个新进队列的节点的 level 值都是父亲节点的 level 值加一。最后根据每个点的 level 对点进行分类,分类的时候我们可以利用哈希表,维护一个以 level 为键,对应节点值组成的数组为值,广度优先搜索结束以后按键 level 从小到大取出所有值,组成答案返回即可。

考虑如何优化空间开销:如何不用哈希映射,并且只用一个变量 node 表示状态,实现这个功能呢?

我们可以用一种巧妙的方法修改广度优先搜索:

首先根元素入队 当队列不为空的时候 求当前队列的长度

s_i
s_i

依次从队列中取

s_i
s_i

个元素进行拓展,然后进入下一次迭代 它和普通广度优先搜索的区别在于,普通广度优先搜索每次只取一个元素拓展,而这里每次取s_i个元素。在上述过程中的第

i
i

次迭代就得到了二叉树的第

i
i

层的

s_i
s_i

个元素。

为什么这么做是对的呢?我们观察这个算法,可以归纳出这样的循环不变式:第

i
i

次迭代前,队列中的所有元素就是第

i
i

层的所有元素,并且按照从左向右的顺序排列。证明它的三条性质(你也可以把它理解成数学归纳法):

初始化:i=1的时候,队列里面只有root,是唯一的层数为1的元素,因为只有一个元素,所以也显然满足「从左向右排列」; 保持:如果i=k时性质成立,即第k轮中出队

s_k
s_k

的元素是第

k
k

层的所有元素,并且顺序从左到右。因为对树进行广度优先搜索的时候由低 k层的点拓展出的点一定也只能是k+1层的点,并且k+1层的点只能由第k层的点拓展到,所以由这s_k个点能拓展到下一层所有的

s_{k+1}
s_{k+1}

个点。又因为队列的先进先出(FIFO)特性,既然第 kkk 层的点的出队顺序是从左向右,那么第k+1层也一定是从左向右。至此,我们已经可以通过数学归纳法证明循环不变式的正确性。 终止:因为该循环不变式是正确的,所以按照这个方法迭代之后每次迭代得到的也就是当前层的层次遍历结果。至此,我们证明了算法是正确的。 代码

代码语言:javascript
复制
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector <vector <int>> ret;
        if (!root) {
            return ret;
        }

        queue <TreeNode*> q;
        q.push(root);
        while (!q.empty()) {
            int currentLevelSize = q.size();
            ret.push_back(vector <int> ());
            for (int i = 1; i <= currentLevelSize; ++i) {
                auto node = q.front(); q.pop();
                ret.back().push_back(node->val);
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
        }
        
        return ret;
    }
};

复杂度分析

记树上所有节点的个数为n。

时间复杂度:每个点进队出队各一次,故渐进时间复杂度为O(n)。 空间复杂度:队列中元素的个数不超过

n
n

个,故渐进空间复杂度为O(n)。

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2024-01-27,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档