广度优先遍历二叉树(BFS)

 

// 广度优先遍历二叉树(BFS).cpp: 定义控制台应用程序的入口点。
//

#include "stdafx.h"


#include <iostream>
#include <stdio.h>
#include <malloc.h>
#include <stdlib.h>

using namespace std;

#define QUEUE_INIT_SIZE 100
typedef char ElemType;

typedef struct BiNode
{
    ElemType data;
    struct BiNode *lchild, *rchild;
}BiNode, *BiPtr;

typedef struct queue
{
    int queuesize;   //数组的大小
    int head, tail;  //队列的头和尾下标
    BiPtr *q;        //数组头指针
}Queue, *QueuePtr;

int InitQueue(QueuePtr &Q)
{
    Q->queuesize = QUEUE_INIT_SIZE;
    Q->q = (BiPtr *)malloc(sizeof(BiPtr) * Q->queuesize); //分配内存
    Q->tail = 0;//数组下标从0开始,tail始终指向队尾元素的下一个位置
    Q->head = 0;//队列非空时，head始终指向队列首元素
    return 0;
}

int EnQueue(QueuePtr Q, BiPtr key)
{
    if((Q->tail + 1) % Q->queuesize == Q->head)//取余保证，当quil=queuesize-1时，再转回0,此时队列没有空间                    
    {
        cout << "队列已满，无法再添加元素！";
    }
    else
    {
        Q->q[Q->tail] = key;
        Q->tail = (Q->tail + 1) % Q->queuesize;
    }
    return 0;
}

BiPtr DeQueue(QueuePtr Q)
{
    BiPtr temp;
    if (Q->tail == Q->head)     //判断队列不为空
    {
        cout << "队列为空，无法删除元素！";
    }
    else
    {
        temp = Q->q[Q->head];
        Q->head = (Q->head + 1) % Q->queuesize;
    }
    return temp;
}

int IsQueueEmpty(QueuePtr Q)
{
    if (Q->head == Q->tail)//空
        return 1;
    else//非空
        return 0;
}

int IsQueueFull(QueuePtr Q)
{
    if ((Q->tail + 1) % Q->queuesize == Q->head)
        return 1;
    else
        return 0;
}

//---------------------------------------------------

int Create_BiTree(BiPtr& T)
{
    ElemType c;
    //cout << "请输入当前节点元素值：" << endl;
    cin >> c;
    if (c == '#') T = NULL;
    else
    {
        T = new BiNode;
        T->data = c;
        Create_BiTree(T->lchild);
        Create_BiTree(T->rchild);
    }
    return 0;
}

//层序遍历
int LevelOrderTraverse(BiPtr T)
{
    QueuePtr Q = new Queue;
    BiPtr p = NULL;
    InitQueue(Q);

    EnQueue(Q, T);//根结点入队
    while (!IsQueueEmpty(Q))
    {
        p = DeQueue(Q);
        cout << p->data << " ";
        if (p->lchild)//如果p有左孩子,则左孩子入队
            EnQueue(Q, p->lchild);
        if (p->rchild)//如果p有右孩子,则左孩子入队
            EnQueue(Q, p->rchild);
    }
    return 0;
}

int main()
{
    freopen("input2.txt", "r", stdin);//从input.txt中读取输入数据
    BiPtr T;
    Create_BiTree(T);
    LevelOrderTraverse(T);
    fclose(stdin);
    return 0;
}

测试方式和上篇相同

第一组数据的遍历输出结果：A B E C D F 

第二组数据的遍历输出结果：A B J C D K E F G I H

两棵树的结构也和上篇完全一样

另外，循环队列的代码借鉴:http://www.cnblogs.com/kaituorensheng/archive/2013/02/28/2937865.html