二叉树 : 层序遍历

二叉树的层序遍历

顾名思义，二叉树的层序遍历就是从左到右一层一层的遍历二叉树，与先序、中序和后序遍历有所不同。

层序遍历是实现需要借助队列。将二叉树的根节点放入队列，然后出队，访问出队结点并判断它是否有左子树和右子树，先入左子树再入右子树；继续出队，出队时访问出队结点是否有左子树…重复步骤直到队列为空，此时整棵树的结点都被遍历完成。

当本层的左子树结点开始出队时，右子树结点也已经再队列中，再入左子树结点的子结点时不会打乱层序的顺序。

代码实现：

//层序遍历
void LevelOrder(BiTree T)
{
    queue<BiTree> q; //借助队列
    if (T != NULL)
    {
        BiTree p;          //暂存要出队的结点
        q.push(T);         //根结点入队  A
        while (!q.empty()) //队列非空
        {
            p = q.front();
            q.pop();       //出队
            printf("%d ", p->data);
            if (p->lchild != NULL)  // B
                q.push(p->lchild); //队列先进先出，先入左孩子 
            if (p->rchild != NULL)  // C
                q.push(p->rchild);
        }
    }
}

leetcode 102：二叉树的层序遍历

力扣的层序遍历要求每一次的结点存储在一个数组中，因此需要使用size存储当前层数下的结点数

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

push_back：
函数将一个新的元素加到vector的最后面，位置为当前最后一个元素的下一个元素
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        if(root != nullptr) q.push(root);
        vector<vector<int>> result;   //结果存储
        while(!q.empty()){
            int size = q.size();  //根据题目要求，记录本层的元素数量
            vector<int> vec;      //存储本层节点的数组
            for(int i = 0; i < size;i++){
                TreeNode* p = q.front(); //取队列的第一个元素
                q.pop();
                vec.push_back(p->val);
                if(p->left != nullptr) q.push(p->left);
                if(p->right != nullptr) q.push(p->right);
            }
            result.push_back(vec);

        }
        return result;
    }
};