数据结构与算法之二叉树的层序遍历

什么是层序遍历

首先，看下图

层序遍历结果应该是：ABCDEFGHIJKL

算法思想

1. 初始化一个辅助队列
2. 根节点入队
3. 若队列非空，则队头节点出队，访问该节点，并将其左、右孩子插入队尾(如果有的话)
4. 重复3直至队列为空

动图展示

代码展示

/**
 * 二叉树的层序遍历
 * @param biTree 二叉树
 */
void levelOrder(BiTree biTree) {
    //初始化一个队列
    BiTNodeSqQueue *queue;
    initBiTNodeSqQueue(&queue);
    //添加根元素之队列
    enBiTNodeSqQueue(queue, *biTree);
    //用于保留出队元素
    BiTNode *deBiNode;
    while (emptyBiTNodeSqQueue(queue)) {
        //队列出队
        deBiTNodeSqQueue(queue, deBiNode);
        printf("%d\t", deBiNode->data);
        //将出队元素的左右子树添加至队列
        if (NULL != deBiNode->lChild) {
            enBiTNodeSqQueue(queue, *deBiNode->lChild);
        }
        if (NULL != deBiNode->rChild) {
            enBiTNodeSqQueue(queue, *deBiNode->rChild);
        }
    }
    printf("%d\t", biTree->data);
}

头文件

BiTNodeSqQueue.h

/**
 * 为了层序遍历BiTNode而做的一个简单队列
 * @author five-five
 * @created 2022/5/30
 *
 */


#include "stdio.h"
#include "stdlib.h"
#include "stdbool.h"
#include "BiTNode.h"

#define MAXSIZE 3 //最大长度
typedef struct {
    BiTNode *data[MAXSIZE];  //静态数组
    int front; //队头
    int rear;  //队尾
    /**
     * 这里说明一点，做这一步是为了不使空间浪费，直接使用rear+1取模运算会浪费一个data[rear]的内存空间
     */
    int tag;  //记录操作，0为入队，1为出队
} BiTNodeSqQueue;

/**
 * 初始化队列
 * @param queue 队列指针的指针
 * @return true?成功:失败
 */
bool initBiTNodeSqQueue(BiTNodeSqQueue **queue) {
    *queue = (BiTNodeSqQueue *) malloc(sizeof(BiTNodeSqQueue));
    if (*queue == NULL) {
        return false;
    }
    (*queue)->front = 0;
    (*queue)->rear = 0;
    (*queue)->tag = 1;
    return true;
}

/**
 * 判断队列是否为空
 * @param queue 要判断的队列
 * @return true?队空:非空
 */
bool emptyBiTNodeSqQueue(BiTNodeSqQueue *queue) {
    if (NULL == queue) {
        return true;
    }
    if (queue->front == 0) {
        return true;
    }
    return false;
}

/**
 * 入队操作
 * @param queue 队列指针
 * @param data 要入队的数据
 * @return true?成功:失败
 */
bool enBiTNodeSqQueue(BiTNodeSqQueue *queue, BiTNode *data) {
    if (queue == NULL) {
        return false;
    }
    int i = (queue->rear + 1) % MAXSIZE;
    if (queue->rear == queue->front && queue->tag == 0) {
        return false;
    }
    queue->data[queue->rear] = data;
    queue->rear = i;
    queue->tag = 0;
    return true;
}

/**
 * 出队操作
 * @param queue 队列指针
 * @param e 用于接收出队元素的指针
 * @return true?成功:失败
 */
bool deBiTNodeSqQueue(BiTNodeSqQueue *queue, BiTNode **e) {
    if (queue == NULL) {
        return false;
    }
    if (queue->rear == queue->front && queue->tag == 1) {
        return false;
    }
    *e = queue->data[queue->front];
    queue->front = (queue->front + 1) % MAXSIZE;
    queue->tag = 1;
    return true;
}

BiTNode.h

/**
 * @author five-five
 * @created 2022/5/30
 *
 */


#include "stdio.h"
#include "stdlib.h"
#include "stdbool.h"

typedef struct BiTNode {
    int data;
    struct BiTNode *lChild;
    struct BiTNode *rChild;
} BiTNode, *BiTree;


/**
 * 初始化结点空间
 * @param biTNode 结点的指针的指针
 * @return true?空间分配成功:空间分配失败
 */
bool initTNode(BiTNode **biTNode) {
    (*biTNode) = (BiTNode *) malloc(sizeof(BiTNode));
    if (NULL == (*biTNode)) {
        false;
    }
    (*biTNode)->lChild = NULL;
    (*biTNode)->rChild = NULL;
    return true;
}

/**
 * 二叉树的前序遍历：根左右
 * @param biTree 二叉树
 */
void preOrder(BiTree biTree) {
    //递归结束
    if (biTree == NULL) {
        return;
    }
    printf("%d\t", biTree->data);
    //左子树，因为子树和根都是一样的数据结构所以可以直接递归
    preOrder(biTree->lChild);
    //右子树，因为子树和根都是一样的数据结构所以可以直接递归
    preOrder(biTree->rChild);
}

/**
 * 二叉树的中序遍历：左根右
 * @param biTree 二叉树
 */
void midOrder(BiTree biTree) {
    //递归结束
    if (biTree == NULL) {
        return;
    }
    //左子树，因为子树和根都是一样的数据结构所以可以直接递归
    midOrder(biTree->lChild);
    //根节点data打印
    printf("%d\t", biTree->data);
    //右子树，因为子树和根都是一样的数据结构所以可以直接递归
    midOrder(biTree->rChild);
}

/**
 * 二叉树的中序遍历：左根右
 * @param biTree 二叉树
 */
void postOrder(BiTree biTree) {
    //递归结束
    if (biTree == NULL) {
        return;
    }
    //左子树，因为子树和根都是一样的数据结构所以可以直接递归
    postOrder(biTree->lChild);
    //右子树，因为子树和根都是一样的数据结构所以可以直接递归
    postOrder(biTree->rChild);
    //根节点data打印
    printf("%d\t", biTree->data);
}