二叉树的遍历以及搜索

最新推荐文章于 2022-05-21 21:31:26 发布

原创最新推荐文章于 2022-05-21 21:31:26 发布 · 484 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#struct #include

C/C++面试题专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种使用C++实现的二叉树数据结构及其基本操作，包括先序、中序、后序和层序遍历。此外，还实现了二叉树的深度计算和节点计数等功能，并通过队列辅助完成层序遍历。

#include <cstdlib>
#include <iostream>
#define ElemType char
#define MaxQsize 20
#define TRUE 1
#define FALSE 0

using namespace std;

typedef struct BiTNode
{
ElemType data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

typedef BiTree QElemType; //设队列元素为二叉树的指针类型

typedef struct
{
  QElemType *base;
  int front; //头指针,若队列不空,指向队头元素
  int rear; //尾指针,若队列不空,指向队尾元素
}SqQueue;

void CreateBiTree_pre(BiTree &T);
void preordertraverse(BiTree T);
void InOrderTraverse(BiTree T);
void postOrderTraverse(BiTree T);
void InitQueue(SqQueue &Q);
bool QueueEmpty(SqQueue Q);
int EnQueue(SqQueue &Q,QElemType &e);
int DeQueue(SqQueue &Q,QElemType &e);
void LevelOrderTraverse(BiTree &T);
void visit(ElemType ch);
int postBiTreeDepth(BiTree T);
int CountNode(BiTree T);

int main()
{
    BiTree T;
    QElemType e;
    int depth,count;
    cout<<"请先序输入二叉树的元素,其中'.'表示空树\n";
    CreateBiTree_pre(T);
    cout<<"先序遍历二叉树为:\n";
    preordertraverse(T);cout<<endl;
    cout<<"中序遍历二叉树为:\n";
    InOrderTraverse(T);cout<<endl;
    cout<<"后序遍历二叉树为:\n";
    postOrderTraverse(T);cout<<endl;
    cout<<"层序遍历二叉树为:\n";
    LevelOrderTraverse(T);cout<<endl;
    cout<<"二叉树的深度为:";
    depth=postBiTreeDepth(T);
    cout<<depth<<endl;
    cout<<"二叉树的节点个数为:";
    count=CountNode(T);
    cout<<count<<endl;
    system("PAUSE");
    return EXIT_SUCCESS;
}

void CreateBiTree_pre(BiTree &T) //先序建立二叉树
{
  char ch;
  cin>>ch;
  if(ch=='.')
  T=NULL;
  else
  {
      if(!(T=(BiTNode*)malloc(sizeof(BiTNode))))
      return;
      T->data=ch;
      CreateBiTree_pre(T->lchild); //构造左子树
      CreateBiTree_pre(T->rchild);  //构造右子树
  }//else
}

void preordertraverse(BiTree T)  //先序遍历二叉树
{
     if(T)
     {
         visit(T->data);
         preordertraverse(T->lchild);
         preordertraverse(T->rchild);
     }
}

void InOrderTraverse(BiTree T)  //中序遍历二叉树
{
     if(T)
     {
       InOrderTraverse(T->lchild);
       visit(T->data);
       InOrderTraverse(T->rchild);
     }
}

void postOrderTraverse(BiTree T)  //后序遍历二叉树
{
     if(T)
     {
          postOrderTraverse(T->lchild);
          postOrderTraverse(T->rchild);
          visit(T->data);
     }
}

void visit(ElemType ch)
{
     cout<<ch<<" ";
}
void InitQueue(SqQueue &Q)   //初始化队列
{
     if(!(Q.base=(QElemType*)malloc(MaxQsize*sizeof(QElemType))))
     return;
     Q.front=Q.rear=0;
     cout<<"初始化成功\n";
}

bool QueueEmpty(SqQueue Q)  //判空
{
     if(Q.front==Q.rear)
     return TRUE;
     else
     return FALSE;
}

int EnQueue(SqQueue &Q,QElemType &e)  //入队
{
    if((Q.rear+sizeof(QElemType))%(MaxQsize*sizeof(QElemType))==Q.front)  //队满
    return 0;
    else
    {
     Q.base[Q.rear]=e;
     Q.rear=(Q.rear+sizeof(QElemType))%(MaxQsize*sizeof(QElemType));
     return 1;
     }
}

int DeQueue(SqQueue &Q,QElemType &e)  //出队
{
    if(QueueEmpty(Q))
    return 0;
    else
    e=Q.base[Q.front];
    Q.front=(Q.front+sizeof(QElemType))%(MaxQsize*sizeof(QElemType));
    return 1;
}

void LevelOrderTraverse(BiTree &T)   //层序遍历二叉树
{
     SqQueue Q;
     BiTree P;
     InitQueue(Q);
     if(T)
     EnQueue(Q,T);
     while(Q.front!=Q.rear)
     {
        DeQueue(Q,P);
        visit(P->data);
        if(P->lchild)
        EnQueue(Q,P->lchild);
        if(P->rchild)
        EnQueue(Q,P->rchild);
     }//end_while
}

int postBiTreeDepth(BiTree T)
{
     int depthval,Leftval=0,Rightval=0;
     if(!T)
     return 0;
     else
     {
          Leftval=postBiTreeDepth(T->lchild);
          Rightval=postBiTreeDepth(T->rchild);
          depthval=1+(Leftval>Rightval?Leftval:Rightval);
         return depthval;
     }//else
}