数据结构--二叉树 (队列实现的层次遍历)

BTree.h:

#include “DS.h”
typedef char ElemType;


typedef struct BiTNode {
    ElemType data;
    struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
typedef struct QueueNode{
    BiTree data;
    struct QueueNode *next;
}QueueNode,*QueuePtr;
typedef struct {
    QueuePtr front;
    QueuePtr rear;
}LinkQueue;

void menu();
void CreateBiTree(BiTree &T);
void PreOrderTraverse(BiTree T);
void InOrderTraverse(BiTree T);
void PostOrderTraverse(BiTree T);
void leverOrderTraverse(BiTree T);
Status InitQueue(LinkQueue *Q);
Status IsEmptyQueue(LinkQueue Q);
Status EnQueue(LinkQueue *Q,BiTree m);
BiTree DeQueue(LinkQueue *Q);

BTree.cpp:

#include “BTree.h”
//菜单
void menu(){
    printf(”\t\t\t 1、创建二叉树\n”);
    printf(”\t\t\t 2、前序遍历\n”);
    printf(”\t\t\t 3、中序遍历\n”);
    printf(”\t\t\t 4、后序遍历\n”);
    printf(”\t\t\t 5、层次遍历\n”);
    printf(”\t\t\t 6、退出\n”);
}
//创建二叉树
void CreateBiTree(BiTree &T){
    ElemType ch;
    scanf(”%c”,&ch);
    if(ch==’#’)
        T=NULL;
    else{
        T=(BiTree)malloc(sizeof(BiTNode));
        if(!T)
            exit(OVERFLOW);
        T->data=ch;
        CreateBiTree(T->lchild);
        CreateBiTree(T->rchild);
        T->data;
    }

}
//前序遍历
void PreOrderTraverse(BiTree T){
    if(T == NULL)
        return ;
    printf(”%c”,T->data);
    PreOrderTraverse(T->lchild);
    PreOrderTraverse(T->rchild);
    return ;
}
//中序遍历
void InOrderTraverse(BiTree T){
    if(T == NULL)
        return ;
    InOrderTraverse(T->lchild);
    printf(”%c ”,T->data);
    InOrderTraverse(T->rchild);
}
//后序遍历
void PostOrderTraverse(BiTree T){
    if(T == NULL)
        return ;
    PostOrderTraverse(T->lchild);
    PostOrderTraverse(T->rchild);
    printf(”%c ”,T->data);

}
//层次遍历
void leverOrderTraverse(BiTree T){
    LinkQueue Q;

    InitQueue(&Q);
    EnQueue(&Q,T);
    while(!IsEmptyQueue(Q)){
        T=  DeQueue(&Q);
        if(T->lchild){
            EnQueue(&Q,T->lchild);
        }
        if(T->rchild){
            EnQueue(&Q,T->rchild);
        }

    }

}
Status InitQueue(LinkQueue *Q){
    Q->front=Q->rear=(QueuePtr)malloc(sizeof(QueueNode));
    if(!Q->front){
        return ERROR;
    }
    Q->front->next=NULL;
    return OK;
}
Status IsEmptyQueue(LinkQueue Q){
    if(Q.front==Q.rear){
        return OK;
    }else{
        return ERROR;
    }
}
Status EnQueue(LinkQueue *Q,BiTree m){
    QueuePtr p=(QueuePtr)malloc (sizeof(QueueNode));
    if(!p)
        return ERROR;
    p->data=m;
    p->next=NULL;
    Q->rear->next=p;
    Q->rear=p;
    return OK;
}
BiTree DeQueue(LinkQueue *Q){
    QueuePtr p;
    BiTree q;
    if(Q->front==Q->rear){
        return ERROR;
    }
    p=Q->front->next;
    if(!p){
        return ERROR;
    }else{
         q= p->data;
         printf(”%c ”,q->data);
        Q->front->next=p->next;
        if(Q->rear==p){
            Q->rear=Q->front;
        }
        free(p);
        return q;
    }
}

main.cpp:

#include “BTree.h”

int main(){
    BiTree T;
    int choice ;
    while(1){
        menu();
        printf(”请输入你的操作:  ”);
        scanf(”%d”,&choice);
        if(choice==6) break;
        switch(choice){
            case 1:
                getchar();
                CreateBiTree(T);
                printf(”二叉树创建完成\n”);
            case 2:
                printf(”二叉树前序遍历的结果是: ”);
                PreOrderTraverse(T);
                printf(”\n”);
                break;
            case 3:
                printf(”二叉树中序遍历的结果是: ”);
                InOrderTraverse(T);
                printf(”\n”);
                break;
            case 4:
                printf(”二叉树后序遍历的结果是: ”);
                PostOrderTraverse(T);
                printf(”\n”);
                break;
            case 5:
                printf(”二叉树层次遍历的结果是: ”);
                leverOrderTraverse( T);
                printf(”\n”);
                break;

        }

    }
   return 0;
}