【力扣】二叉树的分层遍历1和2

简介: 【力扣】二叉树的分层遍历1和2

二叉树的分层遍历1


链接:二叉树的分层遍历1


首先,我们需要知道什么是二叉树的层序遍历。二叉树的层序遍历是一种按照树的层次从上到下,从左到右访问每个节点的方法。例如,对于下面这棵二叉树:

1
   / \
  2   3
 / \ 
4   5


它的层序遍历结果是:[1, 2, 3, 4, 5]。


那么,如何实现这种遍历方法呢?我们可以使用一个队列来辅助我们。队列是一种先进先出(FIFO)的数据结构,也就是说,先进入队列的元素会先出队,后进入队列的元素会后出队。


我们的思路是这样的:


首先,我们判断根节点是否为空,如果为空,就直接返回一个空的二维向量作为结果。

如果不为空,我们就把根节点入队,并记录当前层的节点数为1(因为只有根节点一个节点)。

然后,我们开始一个循环,直到队列为空为止。在每次循环中,我们做以下几件事:

创建一个一维向量来存储当前层的节点值。

使用一个内层循环来遍历当前层的所有节点。在每次内层循环中,我们做以下几件事:

将队首的节点出队,并将其值放入一维向量中。

判断该节点是否有左右子节点,如果有,就将它们入队。

当内层循环结束时,说明当前层的所有节点都已经处理完毕,将一维向量放入二维向量中。

更新下一层的节点数为队列中剩余的元素个数。

当外层循环结束时,说明队列中已经没有元素了,也就是说所有层都已经遍历完毕,返回二维向量作为结果。

这样,我们就实现了二叉树的层序遍历算法。

代码(已加注释):

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) { // 定义一个公共的函数,接受一个二叉树的根节点作为参数,返回一个二维向量,表示每一层的节点值
        queue<TreeNode*> q; // 定义一个队列来存储每一层的节点
        int levelsize =0; // 定义一个变量来记录当前层的节点数
        if(root) // 判断根节点是否为空
        {
            q.push(root); // 如果不为空,就把根节点入队
            levelsize = 1; // 并将当前层的节点数设为1
        }
        vector<vector<int>> vv; // 定义一个二维向量来存储结果
        while(! q.empty()) // 使用一个循环来处理队列中的节点,直到队列为空为止
        {
            //通过levelsize控制一层一层出
            vector<int> v; // 定义一个一维向量来存储当前层的节点值
            while(levelsize --) // 使用一个内层循环来遍历当前层的所有节点
            {
                TreeNode* front = q.front(); // 将队首的节点出队,并保存到一个临时变量中
                q.pop();
                v.push_back(front->val); // 将该节点的值放入一维向量中
                // 带入下一层的节点
                if(front->left) // 判断该节点是否有左子节点
                    q.push(front->left); // 如果有,就将其入队
                if(front->right)  // 判断该节点是否有右子节点 
                 q.push(front->right); // 如果有,就将其入队
            }
            vv.push_back(v); // 当内层循环结束时,说明当前层的所有节点都已经处理完毕,将一维向量放入二维向量中
            //更新下一层的数据
            levelsize = q.size(); // 将下一层的节点数设为队列中剩余的元素个数
        }
        return vv; // 当外层循环结束时,说明队列中已经没有元素了,也就是说所有层都已经遍历完毕,返回二维向量作为结果 
    }
};


二叉树的分层遍历2


链接:层序遍历2 — 自底向上

这个其实很简单,把自上而下遍历的结果逆置一下就行了,复用了

代码:


//....前面一样
        reverse(vv.begin(),vv.end() ); //逆置一下
        return vv; // 当外层循环结束时,说明队列中已经没有元素了,也就是说所有层都已经遍历完毕,返回二维向量作为结果 
    }
};


相关文章
|
2月前
【LeetCode 31】104.二叉树的最大深度
【LeetCode 31】104.二叉树的最大深度
20 2
|
2月前
【LeetCode 29】226.反转二叉树
【LeetCode 29】226.反转二叉树
18 2
|
2月前
【LeetCode 28】102.二叉树的层序遍历
【LeetCode 28】102.二叉树的层序遍历
16 2
|
2月前
【LeetCode 43】236.二叉树的最近公共祖先
【LeetCode 43】236.二叉树的最近公共祖先
20 0
|
2月前
【LeetCode 38】617.合并二叉树
【LeetCode 38】617.合并二叉树
15 0
|
2月前
【LeetCode 37】106.从中序与后序遍历构造二叉树
【LeetCode 37】106.从中序与后序遍历构造二叉树
17 0
|
2月前
【LeetCode 34】257.二叉树的所有路径
【LeetCode 34】257.二叉树的所有路径
20 0
|
2月前
【LeetCode 32】111.二叉树的最小深度
【LeetCode 32】111.二叉树的最小深度
18 0
|
4月前
|
存储 算法
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
文章深入探讨了二叉树的层序遍历方法,并展示了如何通过队列实现层序遍历的算法逻辑,同时指出掌握层序遍历技巧可以帮助解决LeetCode上的多道相关题目。
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
|
4月前
|
算法 Java
LeetCode第94题二叉树的中序遍历
文章介绍了LeetCode第94题"二叉树的中序遍历"的解法,使用递归实现了中序遍历的过程,遵循了"左根右"的遍历顺序,并提供了清晰的Java代码实现。
LeetCode第94题二叉树的中序遍历