【头歌实训：层次遍历二叉树】

头歌实训：层次遍历二叉树

任务描述

本关任务：给定一棵二叉树，借助队列实现层次遍历二叉树。

相关知识

为了完成本关任务，你需要掌握：1.队列基本操作，2.二叉树层次遍历。

队列基本操作

本关卡提供C++ STL模板队列queue的相关操作和功能。

使用实例如下：
queue q; // 创建队列对象
q.push(3); // 元素入队列
q.push(4);
cout<<q.size()<<endl; //打印队列元素个数
while(!q.empty()) {
cout<<q.front()<<endl; // 打印队首元素
q.pop(); // 移除队首元素
}

二叉树层次遍历

层次遍历要求先访问离根节点最近的层的节点，然后依次访问下一层的节点。例如：图1的层次遍历顺序为节点上的数字，结果为：ABECDF。

编程要求

本关的编程任务是补全右侧代码片段LevelOrder中Begin至End中间的代码，具体要求如下：

在LevelOrder中实现二叉树的层次遍历并输出结果，中间没有空格，末尾不换行。

测试说明

平台将自动编译补全后的代码，并生成若干组测试数据，接着根据程序的输出判断程序是否正确。

以下是平台的测试样例：

测试输入：ABC##D##EF###
预期输出：ABECDF

测试输入：ABCD###E#F##G##
预期输出：ABGCEDF

开始你的任务吧，祝你成功！

源代码：

//
//  binary_tree.cpp


#include "binary_tree.h"

// 二叉树的层次遍历（队列实现）
// 参数：二叉树根节点root
// 输出：二叉树的层次遍历，中间没有空格，末尾不换行。
void LevelOrder(BTNode* root)
{
    // 请在这里补充代码，完成本关任务
    /********** Begin *********/
    if(root == NULL) return;    //空树不需要操作
    BTNode *Q[100], *p;     //队列Q用于存储未处理的结点，p用于遍历
    int rear = 0, head = 0; //队列置为空
    p = root;   //开始遍历
    while(p || rear != head)    //当结点非空或者队列不空开始遍历
    {
        if(rear == head)    //如果开始是空队列直接入队列
        {
            Q[rear] = p;    //入队列
            rear = (rear + 1) % 100;    //循环队列队尾这样子增
        }
        else
        {
            p = Q[head];    //队列不空获得队头元素
            head = (head + 1) % 100;    //队头指针加
            cout << p->data;    //输出队头元素
            if(p->lchild)   //如果他有左孩子左孩子入队
            {
                Q[rear] = p->lchild;
                rear = (rear + 1) % 100;
            }
            if(p->rchild)   //有右孩子右孩子入队
            {
                Q[rear] = p->rchild;
                rear = (rear + 1) % 100;
            }
            p = NULL;   //p置为空为了判断循环条件
        }
        //if(rear == head && p == NULL) break;
    }

    /********** End **********/
}