二叉树层次遍历的进阶

本文原载于我的博客：https://www.seekingmini.top/archives/二叉树层次遍历的进阶

写在前面

在教科书上学到的关于二叉树的层次遍历，只是简单的按照每一层打印元素。如果想要区分每一层，应该怎么做？

简要介绍

定义

按照从上到下、从左到右的规则，依次遍历二叉树的元素。

二叉树如下：
     5
   /   \
  7     9
 / \   / \
1   3 2   6
 
层次遍历的顺序：
5 -> 7 -> 9 -> 1 -> 3 -> 2 -> 6

算法步骤

1. 初始化一个队列，并将根结点置入队列中；
2. 打印队首元素值后将其弹出；
3. 将队首元素的左右子结点分别置入队列中（如果没有子结点就跳过）；
4. 重复步骤2、3直到队列为空。

代码

#include <queue>
#include <cstdio>
using namespace std;

struct Node
{
    char val;
    Node *lchild;
    Node *rchild;
    Node()
    {
        this->val = '#';
        this->lchild = NULL;
        this->rchild = NULL;
    }
};

void BFS(Node *root)
{
    if (root == NULL)
        return;
    queue<Node *> q;
    q.push(root);

    while (!q.empty())  // 队列不为空
    {
        Node *node = q.front();  // 队首元素
        q.pop();
        printf("%c ", node->val);
        if (node->lchild)
            q.push(node->lchild);
        if (node->rchild)
            q.push(node->rchild);
    }
}

使用队列实现层次遍历的原理是：每pop出一个结点，就把这个结点的子结点放入队列中，直到队列为空。

拓展

上面的代码是层次遍历的一个正确实现，但是不够直观——我们无法得知哪些结点属于哪一层。那么怎样对算法进行改进，可以正确反映结点的层次呢？

二叉树如下：
     5
   /   \
  7     9
 / \   / \
1   3 2   6
由于要反映各个结点的层次，我们假设返回的结果是：
[[5], 
 [7, 9], 
 [1, 3, 2, 6]]
对于这样一个"不定长"的二维数组，行数即是当前结点的层次。

普通的层次遍历是一轮循环中只pop出一个结点。如果要将各个层次区分开，我们使用一个内循环，把队列中的每一个结点的子结点依次push进队列，然后再将父结点pop。举个例子：

以上面的二叉树为例，队列初始状态只有根结点3。
第1轮循环，此时队列长度为1，记为current_num，然后内循环1次，将3的子结点7和9加入队列再pop出3；
第2轮循环，此时队列长度为2，记为current_num，然后内循环2次，将7和9的子结点1、3、2、6加入队列再pop出7和9；
第3轮循环，此时队列长度为3，记为current_num，然后内循环4次，将1、3、2、6的子结点（也可以没有）加入队列再pop出1、3、2、6；
第4轮循环，此时队列为空，结束循环。

代码

#include <vector>
#include <queue>
using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

vector<vector<int>> levelOrderBottom(TreeNode *root)
    {
        vector<vector<int>> res;
        if (!root)
        {
            return res;
        }
        queue<TreeNode *> q;
        q.push(root);
        while (!q.empty())
        {
            int current_num = q.size(); // 当前队列元素个数（即二叉树当前层结点数 ）
            vector<int> one_level;
            for (; current_num > 0; current_num--)
            {
                TreeNode *f = q.front();
                q.pop();
                one_level.push_back(f->val);
                if (f->left)
                    q.push(f->left);
                if (f->right)
                    q.push(f->right);
            }
            res.push_back(one_level);
        }
        return res;
    }

总结

把两种层次遍历的方法进行对比来看，普通的层次遍历就有了两种方法：第一种就是一次只pop出一个结点再将这个结点的子结点加入队列，第二种就是一次性pop一层的结点再将每个结点的结点分别加入队列。