二叉树的层序遍历

最新推荐文章于 2023-07-10 23:50:03 发布

继续拉扯

最新推荐文章于 2023-07-10 23:50:03 发布

阅读量1.8k

点赞数 4

文章标签： c++ 数据结构广度优先

本文链接：https://blog.csdn.net/qq_61345460/article/details/128132975

版权

前言

我们知道二叉树的遍历有，前中后序遍历，以及层次遍历，前序遍历是以根左右顺序进行遍历，中序遍历是以左根右的顺序进行遍历，而后序遍历是以左右根的顺序进行遍历。其实前中后序遍历的前中后主要是依据根遍历的顺序而言的！我们可以以此进行记忆。前中后序遍历可以使用递归的方式实现，也可以使用非递归也就是栈的辅助实现，而层序遍历我们则要借助队列来实现！

一、二叉树的层序遍历

所谓的层序遍历，顾名思义，就是指一层一层的遍历，如图所示的一棵二叉树：

它层序遍历的结果就是A,B,C,D,E,F,G。很明显栈先进后出适合模拟深度优先遍历，也就是递归的逻辑，无法帮助我们实现层序遍历。因此，我们需要借用另一个辅助数据结构即队列来实现，队列先进先出，符合一层一层遍历的逻辑。

二、方法步骤

那么我们如何用对列模拟实现呢？如图所示的另一棵二叉树：

我们先进行第一层遍历，也就是将A添加到队列中，队列如下所示：

接着我们将A取出，保存到一个vector数组res中，紧接着将 A的左右节点B,C添加到队列中，队列如下所示：

这样我们的第一层遍历就结束，此时我们的res数组中存的是A，由于最终的遍历结果需要保存到一个二维的vector数组result中，因此我们还需将第一层遍历的结果res添加到result中。
然后我们进行第二层的遍历，我们需要将B取出存放到res数组中，接着将B的左右节点也就是D,E添加到队列中，队列如下所示：

再然后我们将C取出存放到res数组中，然后将C的的左右节点添加到队列中，由于C没有左节点，我们只要将F添加到队列中即可，队列如下所示：

这样我们的第二层遍历就结束了，此时我们的res数组中存放的是B,C，我们将其加入到result中。

最后我们只需要如此反复即可实现二叉树的层序遍历！

三、代码演示

我以LeetCode的第102题为例，演示一下代码，题目如下：

完整代码如下：

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

每一次循环我们都要以size来记住当层元素的个数，也就是当前队列的长度，避免循环出错！