当前位置 博文首页 > C语言实现二叉搜索树的完整总结

    C语言实现二叉搜索树的完整总结

    作者:奋斗的龙猫 时间:2021-06-27 17:46

    1、 二叉树的构建

    我们都知道二叉搜索树的特点是:当前节点的值大于它的左子树的值,小于等于右子树的值。所以我们这里可以通过迭代的方式构建二叉搜索树,当然也可以通过递归的方式构建二叉树。

    定义一个结构体,表示节点:

    typedef struct NODE{
        int va;
        struct NODE *left,*right;
    }Node;
    

    ①通过迭代的方式实现二叉搜索树的构建,值得注意的是,这种方式构建二叉搜索树的时候,需要定义一个变量,表示这个节点插入的位置是父节点的左子节点还是右子节点的位置,同时定义一个变量,表示插入的父节点。

    Node * createBinaryTree(Node *root,int val){
        int isLeftChild = 0;//定义一个临时变量,表示这个新节点的插入位置是否为它的左子节点
        Node *cur = root,*parent = NULL,*node;
        node = (Node *)malloc(sizeof(Node));
        if(node == NULL){
           printf("创建节点失败!!!\n");
           exit(0);//退出虚拟机
        }
        node->val = val;
        node->left = node->right = NULL;
        while(*cur != NULL){
         //找到新节点要插入的位置
           parent = cur;
           if(cur->val > x){
              cur = cur->left;//新节点的值小于当前节点的值,那么就往当前节点的左子树方向进行查找
              isLeftChild = 1;
          } else{
              cur = cur->right;//如果新节点的值大于等于当前节点的值,那么就往当前节点的右子树方向进行查找
              isLeftChild = 0;
          }
        }
       //判断parent/root是否为空,如果为空,说明新节点是根节点
       if(pre == NULL){
         root = node;
       }else{
         //parent不为空,说明不是空树,这是需要判断插入的位置是否是在左子节点的位置
         if(isLeftChild){
             parent->left = node;
         }else{
             parent->right= node;
          }
       }
       return root;
    }
    

    ②通过迭代的方式进行创建二叉搜索树

    Node *createBinaryTree(Node *root,int val){
         if(root == NULL){
              root = (Node *)malloc(sizeof(Node));//给新节点分配空间
              if(root == NULL){
                 printf("创建节点失败!!!\n"):
                 exit(0);//退出虚拟机
              }
              root->val = val;
              root->left = root->right = NULL;
          }else{
          //如果当前的节点不为空,那么就判断新节点插入的是左子节点还是右子节点的位置
             if(val < root->val)//新节点的值小于当前节点的值,说明将其插入在当前节点左子树的位置
                root->left = createBinaryTree(root->left,val);
             else//新节点的值大于等于当前节点的值,说明时将其插入在当前节点的右子树位置
                root->right = createBinaryTree(root->right,val);
          }
          return root;
    }
    

    2、二叉树的遍历

    二叉树的遍历主要包括几种遍历方式,分别是前序遍历,中序遍历,后序遍历,层序遍历。
    前序遍历:先访问当前的节点,然后访问它的左子树,最后访问它的右子树。
    中序遍历:先访问当前节点的左子树,然后访问自身,最后访问它的右子树。
    后序遍历:先访问当前节点的左子树,然后访问当前节点的右子树,最后才访问自身。
    层序遍历:一层一层,从左到右遍历的。

    前序遍历

    递归实现

    void preOrderDisplay(Node *root){
       if(root != NULL){
           printf("%5d",root->val);//访问自身
           preOrderDisplay(root->left);//访问当前节点的左子树
           preOrderDisplay(root->right);//访问当前节点的右子树
       }
    }
    

    迭代实现

    注意的是,通过迭代实现二叉树的前序遍历,我们需要利用到栈。

    void preOrderTraversal(Node *root){
      Stack *s;
      if(!createStack(s)){
        printf("创建栈失败!!!\n");
        return;
      }
      Node *t = root,k;
      while(t != NULL || !isEmpty(s)){
        //当前的节点不为空,或者栈不为空,那么就继续进循环
        while(t!= NULL){
            //如果当前的节点不为空,那么就将当前的节点输出,然后将它的左子树压入栈中(遍历到最左)
            printf("%5d",t->val);//由于是前序遍历,那么先输出父节点的值
            pushStack(s,t);
            t = t->left;
        }
        if(!isEmpty(s)){
            //如果栈不为空,那么这时候,将从栈中跳出一个节点,并且将获得它的右子树,然后将右子树压入栈中
            popStack(s,k);//(跳出一个节点)
            t = k.right;//将右子树重复上面的操作(往这个跳出节点k的右子树方向移动)
        }
      }
    }
    

    中序遍历

    递归实现

    //利用递归中序遍历树
    void InOrderDisplay(Node *root){
       if(root != NULL){
        //如果节点不为空,那么递归实现中序遍历
           InOrderDisplay(root->left);//先访问左子树
           printf("%5d",root->val);//访问自身
           InOrderDisplay(root->right);//访问右子树
       }
    }
    

    迭代实现

    /*
    利用迭代循环实现树的中序遍历
    基本思路:利用堆栈实现的
    基本步骤:
    1、判断当前的节点或者栈是否为空,如果其中至少有一个不为空,那么
    这时候将进入循环
    2、判断当前的节点是否为空,(必须要判断,因为进入外部循环的循环条件有两个,所以不知道是否因为当前
    节点是否为空),如果节点不为空,那么将当前的节点压入栈中,然后当前的节点变成它的左节点,将它的左子树压入
    栈中
    3、判断栈是否为空,将栈顶节点跳出,并将其输出,然后后去这个跳出节点的右子节点
    
    */
    void InOrderTraversal(Node *root){
       Stack *s;
       Node *t = root,k;
       if(!createStack(s)){
          printf("创建栈失败!!!\n");
          return;
       }
       while(t != NULL || !isEmpty(s)){
          while(t != NULL){
             pushStack(s,t);//将当前的节点及其左子树压入栈中(遍历到最左)
             t = t->left;
          }
          if(!isEmpty(s)){
            //从栈中跳出最后一个左子节点的父节点
            popStack(s,k);
            printf("%5d",k.val);//输入当前节点的值
            t = k.right;//将其右子树压入栈中(往跳出节点k的右子树方向移动)
          }
       }
    }
    

    后序遍历

    递归实现

    /*
    递归实现树的后序遍历
    */
    void postOrderDisplay(Node *root){
        if(root != NULL){
            //当前的节点不为空,那么就先访问左子树,然后访问右子树,最后访问当前的节点
            postOrderDisplay(root->left);
            postOrderDisplay(root->right);
            printf("%5d",root->val);
        }
    }
    

    迭代实现

    /*
    利用迭代实现树的后序遍历:
    基本思路:
    1、判断当前的节点或者栈是否为空,如果其中至少有一个不为空,那么循环继续
    2、判断该当前的节点是否为空,如果不为空,那么就将当前的节点及其左子树压入栈中
    3、判断当前的栈是否为空,如果不为空,那么就从栈中跳出一个节点
    获取这个节点的右子节点,如果这个右子节点为空,那么就将当前的节点输出,然后再吃从栈中跳出一个节点
    4、重复上面的2、3步骤
    */
    void postOrderTraversal(Node *root){
       Node *t = root,k,pre;//pre表示上一次访问过的右子节点
       Stack *s;
       if(!createStack(s)){
        printf("创建栈失败!!!\n");
        return;
       }
       while(t != NULL || !isEmpty(s)){
        //如果当前的节点不为空或者栈不为空,那么就继续循环遍历
         while(t != NULL){
            //如果当前的节点不为空,那么就将其压入栈中
            pushStack(s,t);
            t = t->left;
         }
         //注意这里并不是直接从栈中跳出一个节点,而是先获取栈顶节点,判断条件满足之后才跳出节点
         if( getTop(s,k) && k.right == NULL || pre.val == k.right->val){
           /*
           判断当前的栈顶节点的右子节点是否为空,或者这个栈顶的右子节点已经输
           出过了,如果这个栈顶节点的右子节点为空或者已经输出过了,那么就将这
           个栈顶节点从栈中跳出,并输出它的值否则,就将这个栈顶节点的右子树压
           入栈中,重复循环操作
           */
            popStack(s,k);
            pre = k;
            printf("%5d",k.val);
         }else{
            t = k.right;//如果上面的条件不满足,那么就往它的右子树方向移动
         }
       }
    }
    

    测试完整代码:

    #include<stdio.h>
    #include<stdlib.h>
    #define MAX_SIZE 100
    #define INCREMENT 10
    #define ERROR 0
    #define OK 1
    typedef struct NODE{
        int val;
        struct NODE *left;
        struct NODE *right;
    }Node;
    typedef struct STACK{
        Node * arr;
        int top;
    }Stack;
    //创建栈
    int createStack(Stack *s){
        s->arr = (Node *)malloc(sizeof(Node) * MAX_SIZE);//分配MAX_SIZE个空间
        if(s->arr == NULL)
            //如果arr为空,说明分配空间事变,这时候返回ERROR
            return ERROR;
        s->top = 0;
        return OK;
    }
    //压栈
    int pushStack(Stack *s,Node *node){
        if(s->top == MAX_SIZE){
            return ERROR;
        }
        Node t;
        t.val = node->val;
        t.left = node->left;
        t.right = node->right;
        s->arr[s->top++] = t;
        return OK;
    }
    //出栈
    int popStack(Stack *s,Node &node){
        if(s->top == 0){
            //如果栈为空,那么这时候返回ERROR
            return ERROR;
        }
        node = s->arr[--s->top];//获取栈顶节点
        return OK;
    }
    int getTop(Stack *s,Node &k){
        if(s->top == 0)
            return ERROR;
        k = s->arr[s->top - 1];
        return OK;
    }
    //判断栈是否为空
    int isEmpty(Stack *s){
        return s->top == 0;
    }
    /*
    节点的插入基本思路:
    判断这颗树是否为空树,如果是一棵空树,那么新节点就是整棵树的
    根节点,如果不是,那么就需要通过遍历找到插入的位置。
    根据二叉搜索树的特点,如果新节点的值小于根节点或者父节点的值,那么就
    往左边走,找到第一个为空的地方,然后将其插入;如果新节点的值大于等于父节点的值,
    那么就往右边走,找到第一个为空的地方,将其插入。
    值得注意的是,我们需要标记插入的是否为左子节点还是右子节点,所以需要定义一个临时
    变量,判断插入的位置是否为父节点的左节点
    */
    Node * insert(Node *root,int val){
       Node *node = (Node *)malloc(sizeof(Node));
       node->val = val;
       node->left = NULL;
       node->right = NULL;
      //如果不是空树,那么就需要定义临时变量,表示插入的位置是否为左节点
      //同时定义一个临时节点,表示要插入位置的父节点
      Node *current = root,*parent = NULL;
      int isLeftChild = 1; //值为1表示插入的是父节点的左子节点的位置,否则为右子节点的位置
      while(current != NULL){
         parent = current;//表示插入位置的父节点
         if(current->val > val){
            //如果当前的节点比新节点的值大,那么就往左子节点的方向走
            isLeftChild = 1;
            current = current->left;
         }else{
            isLeftChild = 0;
            current = current->right;
         }
      }
      if(parent == NULL){
        //如果parent为空,说明是一棵空树,此时新节点就是根节点
        root = node;
      }else{
        if(isLeftChild)
            parent->left = node;
        else
            parent->right = node;
      }
      return root;
    }
    //利用递归中序遍历树
    void InOrderDisplay(Node *root){
       if(root != NULL){
        //如果节点不为空,那么递归实现中序遍历
           InOrderDisplay(root->left);//先访问左子树
           printf("%5d",root->val);//访问自身
           InOrderDisplay(root->right);//访问右子树
       }
    }
    void preOrderDisplay(Node *root){
       if(root != NULL){
        //如果root节点不为空,那么就进行递归
          printf("%5d",root->val);
          preOrderDisplay(root->left);//访问左子树
          preOrderDisplay(root->right);//访问右子树
       }
    }
    /*
    递归实现树的后序遍历
    */
    void postOrderDisplay(Node *root){
        if(root != NULL){
            //当前的节点不为空,那么就先访问左子树,然后访问右子树,最后访问当前的节点
            postOrderDisplay(root->left);
            postOrderDisplay(root->right);
            printf("%5d",root->val);
        }
    }
    /*
    利用迭代实现树的后序遍历:
    基本思路:
    1、判断当前的节点或者栈是否为空,如果其中至少有一个不为空,那么循环继续
    2、判断该当前的节点是否为空,如果不为空,那么就将当前的节点及其左子树压入栈中
    3、判断当前的栈是否为空,如果不为空,那么就从栈中跳出一个节点
    获取这个节点的右子节点,如果这个右子节点为空,那么就将当前的节点输出,然后再吃从栈中跳出一个节点
    4、重复上面的2、3步骤
    */
    void postOrderTraversal(Node *root){
       Node *t = root,k,pre;//pre表示上一次访问过的右子节点
       Stack *s;
       if(!createStack(s)){
        printf("创建栈失败!!!\n");
        return;
       }
       while(t != NULL || !isEmpty(s)){
        //如果当前的节点不为空或者栈不为空,那么就继续循环遍历
         while(t != NULL){
            //如果当前的节点不为空,那么就将其压入栈中
            pushStack(s,t);
            t = t->left;
         }
         //注意这里并不是从栈中跳出一个节点
         if( getTop(s,k) && k.right == NULL || pre.val == k.right->val){
           /*
           判断当前的栈顶节点的右子节点是否为空,或者这个栈顶的右子节点已经输出过了
           如果这个栈顶节点的右子节点为空或者已经输出过了,那么就将这个栈顶节点从栈中跳出,并输出它的值
           否则,就将这个栈顶节点的右子树压入栈中,重复循环操作
           */
            popStack(s,k);
            pre = k;
            printf("%5d",k.val);
         }else{
    
            t = k.right;
         }
       }
    }
    /*
    利用迭代循环实现树的中序遍历
    基本思路:利用堆栈实现的
    基本步骤:
    1、判断当前的节点或者栈是否为空,如果其中至少有一个不为空,那么
    这时候将进入循环
    2、判断当前的节点是否为空,(必须要判断,因为进入外部循环的循环条件有两个,所以不知道是否因为当前
    节点是否为空),如果节点不为空,那么将当前的节点压入栈中,然后当前的节点变成它的左节点,将它的左子树压入
    栈中
    3、判断栈是否为空,将栈顶节点跳出,并将其输出,然后后去这个跳出节点的右子节点
    
    */
    void InOrderTraversal(Node *root){
       Stack *s;
       Node *t = root,k;
       if(!createStack(s)){
          printf("创建栈失败!!!\n");
          return;
       }
       while(t != NULL || !isEmpty(s)){
          while(t != NULL){
             pushStack(s,t);//将当前的节点及其左子树压入栈中
             t = t->left;
          }
          if(!isEmpty(s)){
            //从栈中跳出最后一个左子节点的父节点
            popStack(s,k);
            printf("%5d",k.val);
            t = k.right;//将其右子数压入栈中
          }
    
       }
    }
    /*
    前序遍历的非递归实现:
    基本思路:利用栈实现的
    1、如果当前节点不为空或者当前栈不为空,那么就进入循环语句
    2、如果当前的节点不为空,那么这时候将当前的节点输出,然后将当前节点压入栈中
    然后这个节点往它的左子节点的方向移动,重复2的步骤,知道左子节点为空
    3、如果栈不为空,那么就从栈中跳出一个节点,然后将往这个节点的右子树方向移动
    4、重复上面的2、3步骤
    */
    void preOrderTraversal(Node *root){
      Stack *s;
      if(!createStack(s)){
        printf("创建栈失败!!!\n");
        return;
      }
      Node *t = root,k;
      while(t != NULL || !isEmpty(s)){
        //当前的节点不为空,或者栈不为空,那么就继续进循环
        while(t!= NULL){
            //如果当前的节点不为空,那么就将当前的节点输出,然后将它的左子树压入栈中
            printf("%5d",t->val);//由于是前序遍历,那么先输出父节点的值
            pushStack(s,t);
            t = t->left;
        }
        if(!isEmpty(s)){
            //如果栈不为空,那么这时候,将从栈中跳出一个节点,并且将获得它的右子树,然后将右子树压入栈中
            popStack(s,k);
            t = k.right;//将右子树重复上面的操作
        }
      }
    }
    int main(){
      int n,i,val;
      Node *root = NULL;
      printf("请输入树的节点个数:");
      scanf("%d",&n);
      printf("请输入各个节点的值:");
      for(i = 0; i < n; i++){
        scanf("%d",&val);
        root = insert(root,val);
      }
      printf("递归实现树的中序遍历:");
      InOrderDisplay(root);
      printf("\n");
      printf("迭代实现数的中序遍历:");
      InOrderTraversal(root);
      printf("\n");
      printf("递归实现树的前序遍历:");
      preOrderDisplay(root);
      printf("\n");
      printf("迭代实现树的前序遍历:");
      preOrderTraversal(root);
      printf("\n");
      printf("递归实现树的后序遍历:");
      postOrderDisplay(root);
      printf("\n");
      printf("迭代实现树的后序遍历:");
      postOrderTraversal(root);
      printf("\n");
      return 0;
    }

    运行结果:

    在这里插入图片描述

    层序遍历

    二叉搜索树的层序遍历,需要使用到队列。
    基本思路:
    1·、定义一个队列
    2、创建二叉搜索树
    3、将当前的根节点压入到队列中
    4、当队列不为空的时候,那么我们将从队列中跳出节点,将它的值输出,然后判断它的左右子节点是否为空,如果不为空,那么我们就将他们压入到队列中
    5、重复4的操作,直到队列为空,此时层序遍历完成。
    代码实现:

    /*
    实现二叉树的层序遍历基本思路:
    利用队列来实现的
    1、判断当前的节点是否为空或者队列是否为空,如果
    不为空,那么就将当前的节点压入队列,同时需要判断当前
    节点的子节点是否为空,如果不为空,那么同样的将它的子节点压入队列中
    2、如果把这个节点的子节点压入道队列之后,那么这时候我们需要将从
    队列中跳出一个节点,然后将这个节点的信息输出。
    3、获取队列头,如果队列头不为空,那么这时候重复2的操作
    */
    #include<stdio.h>
    #include<stdlib.h>
    #define MAX_SIZE 100
    #define ERROR 0
    #define OK 1
    typedef struct NODE * Node;
    typedef Node * List;
    struct NODE{
       int val;
       Node left;
       Node right;
    };
    typedef struct QUEUE{
       List arr;
       int front;//队头指针
       int rear;//队尾指针
    }Queue;
    int init(Queue &s){
      s.arr = (List)malloc(sizeof(List) * MAX_SIZE);//定义一个指针类型的数组
      if(s.arr == NULL){
        return ERROR;
      }
    
      int i;
      //给数组初始化之后还没有可以,还需要给所有的节点分配空间,如果没有这一步,那么就会发生报错
      for(i = 0; i < MAX_SIZE; i++){
        s.arr[i] = (Node)malloc(sizeof(struct NODE));
        if(s.arr[i] == NULL)
            return ERROR;
      }
      s.front = s.rear = 0;//将队头指针、队尾指针都初始为0
      return OK;
    }
    //压入队列
    int pushQueue(Queue &s,Node &node){
       if((s.rear + 1) % MAX_SIZE == s.front){
        //如果栈满了,返回ERROR
        printf("队列为满!!!\n");
        return ERROR;
       }
       s.arr[s.rear] = node;
       s.rear = (s.rear + 1) % MAX_SIZE;
       return OK;
    }
    int popQueue(Queue &s,Node &k){
       if(s.rear == s.front){
         //printf("队列为空!!!\n");
         return ERROR;
       }
       k = s.arr[s.front];
       s.front = (s.front + 1) % MAX_SIZE;
       return OK;
    }
    int getTop(Queue &s,Node &k){
       if(s.rear == s.front){
         //printf("队列为空!!!\n");
         return ERROR;
       }
       k = s.arr[s.front];
       return OK;
    }
    int isEmpty(Queue &s){
       return s.rear == s.front;//判断队列是否为空
    }
    int getSize(Queue &s){
       return (s.rear - s.front + MAX_SIZE)%MAX_SIZE;//获取队列的个数
    }
    /*
    利用递归创建二叉查找树
    基本思路:
    1、首先判断当前的节点是否为空,如果为空,就说明这个位置是新节点要插入的位
    置此时需要给新节点分配空间,判断创建节点是否成功,如果失败,那么输出错误信
    息,否则将这个节点返回
    2、如果当前的节点不为空,那么这时候拿当前节点和新节点的值进行比较,如果
    新节点的值大于等于当前的节点,那么意味着新节点会插入在当前节点的右子树位
    置,否则插入在当前节点的左子树位置
    */
    Node createBinaryTree(Node root,int x){
       if(root == NULL){
          Node node = (Node)malloc(sizeof(struct NODE));
          if(node == NULL){
            //printf("创建新节点失败!!!\n");
            exit(0);
          }
          node->val = x;
          node->left = NULL;
          node->right = NULL;
          root = node;
       }else{
          //如果当前的节点不为空,说明不是要插入的位置,需要和当前节点的值进行
          //比较,如果大于等于当前节点的值,那么往右子树的方向进行递归,否则往左子树方向递归
          if(x < root->val){
            root->left = createBinaryTree(root->left,x);
          }else{
            root->right = createBinaryTree(root->right,x);
          }
       }
       return root;
    }
    /*
    利用递归实现树的后序遍历
    */
    void postOrderTraversal(Node root){
       if(root != NULL){
        //如果当前的节点不为空,那么就先访问左子树,然后访问右子树,最后访问自身
          postOrderTraversal(root->left);
          postOrderTraversal(root->right);
          printf("%5d",root->val);
       }
    }
    
    /*
    利用递归实现树的前序遍历
    */
    void preOrderTraversal(Node root){
       if(root != NULL){
          printf("%5d",root->val);
          preOrderTraversal(root->left);
          preOrderTraversal(root->right);
       }
    }
    /*
    利用队列实现树的层序遍历
    */
    void levelOrderTraversal(Node root){
         Node t = root,k;
         Queue q;
         init(q);
         pushQueue(q,t);//将根节点压入队列中
         while(!isEmpty(q)){
            //如果队列不为空,那么就继续进行循环 
            popQueue(q,t);//将从队列中跳出一个节点,然后将这个节点的信息输出
            printf("%5d",t->val);
            /*
            判断从队列中跳出的节点是否含有左右子节点,如果含有,那么就将这个节
            点的左右子节点压入到队列中
            */
            if(t->left != NULL){
                pushQueue(q,t->left);
            }
            if(t->right != NULL){
                pushQueue(q,t->right);
            }
         }
    }
    /*
    为了使层序遍历看的更加直观,我们将定义一个临时变量size,表示在压入队列之前
    队列的元素个数,然后将队列中的元素不断跳出,并且输出对应的信息,与此同时,
    每跳出一个节点,我们都需要判断这个节点是否含有左右子节点,如果含有,那么就
    将它的子节点压入到队列中去
    */
    void levelOrderTraversal2(Node root){
         Node t = root,k;
         Queue q;
         int size,i;
         init(q);
         pushQueue(q,t);//将根节点压入队列中
         while(!isEmpty(q)){
            size = getSize(q);
            for(i = 1; i <= size; i++){
                popQueue(q,k);
                printf("%5d",k->val);
                //每跳出一个节点,那么就将它的左右子节点压入到队列中
                if(k->left != NULL){
                    pushQueue(q,k->left);
                }
                if(k->right != NULL){
                    pushQueue(q,k->right);
                }
            }
            printf("\n");
         }
    }
    
    int main(){
      int n,i,val;
      printf("请输入节点个数:");
      scanf("%d",&n);
      printf("请输入各个节点的值:");
      Node root = NULL;
      //创建二叉查找树
      for(i = 0; i < n; i++){
        scanf("%d",&val);
        root = createBinaryTree(root,val);
      }
      //实现它的后序遍历
      printf("递归实现树的后序遍历:");
      postOrderTraversal(root);
      printf("\n递归实现树的前序遍历:");
      preOrderTraversal(root);
      printf("\n实现树的层序遍历:");
      levelOrderTraversal(root);
      printf("\n递归实现树的层序遍历2\n:");
      levelOrderTraversal2(root);
      return 0;
    }

    运行结果:

    在这里插入图片描述

    4、二叉树的高度

    求解二叉树某一个节点的高度的时候,我们需要获得这个节点的左右子树的高度,然后将两者中的最大值加1就是当前这个节点的高度.
    对应的代码:

    //节点
    typedef struct NODE{
        int val;
        struct NODE *left;
        struct NODE *right;
    }Node;
    int getHeight(Node * root){
        int hl = 0,hr = 0,max;//hl表示的使左子树的高度,hr表示的使右子树的高度
        if(root != NULL){
           //当前的节点不为空,获取左右子树的高度
           hl = getHeight(root->left);
           hr = getHeight(root->right);
           max = hl > hr ? hl : hr;
           return max + 1;//左右子数高度的最大值加1就是当前节点的高度
        }else return 0;//如果当前节点为空,那么它的高度为0
    }
    

    5、二叉树的删除

    二叉搜索树的删除需要考虑三种情况:删除的节点是一个叶子节点、是一个含有一个子节点的节点、是一个含有两个子节点的节点。需要综合这三种情况进行书写代码。

    Node deleteElement(Node root,int x){
       if(root == NULL){
         printf("节点为空,无法进行删除操作!!!");
       }else if(x < root->val){
          root->left = deleteElement(root->left,x);
       }else if(x > root->val){
            root->right = deleteElement(root->right,x);
        }else{
          /*如果当前的节点是要删除的节点
          判断这个删除的节点是否为一个叶节点,如果是,那么直接将其变成NULL即可
          否则,如果这个删除节点只有一个子节点,那么就将子节点的值赋值给这个删
          除节点,然后将它的子节点变成为NULL,否则,如果这个删除节点含有两个子节点,那么
          就将遍历它的右子树,获取右子树中的最小值,然后将这个右子树的最小值赋值给这个
          删除节点的值,在将这个最小值变成NULL
          */
              if(root->left != NULL && root->right != NULL){
                //删除节点含有两个子节点
                Node tmp = findMin(root->right);
                root->val = tmp->val;
                root->right = deleteElement(root->right,tmp->val);
              }else{
                 /*
                 下面的代码如果使这样写的话,会发生错误的,为什么会这样呢?
                 其实很简单,因为这里已经包括了两种情况了,删除的节点是一个叶
                 节点或者只有一个子节点的节点,如果是这样写的话,并没有解决删
                 除节点是一个叶节点的情况,只是把这个删除节点的内存空间释放了
                   Node *t = root;
                 if(root->left != NULL){
                    root = root->left;
                 }else if(root->right != NULL){
                    root = root->right;
                 }
                 free(t);//释放删除的节点
                 */
                 Node t = root;
                 if(root->left == NULL){
                 /*
                 如果当前节点的左子节点为空,那么就用它的右子节点替换当前节
                 点,否则用左子节替换,这样进行判断的好处就是,如果这个删除节点
                 是一个叶节点,那么两个子节点都是空的,那么这时候root = root-
                 >right = NULL了,如果这个删除节点含有一个子节点,并且它的左
                 子节点为空,那么这个节点就用它的右子节点替换,下面的if判断同
                 理
                 */
                    root = root->right;
                 }else if(root->right == NULL){
                    root = root->left;
                 }
                 free(t);//释放删除的节点
    
              }
          }
       return root;
    }

    6、由几种遍历序列还原二叉树

     前序序列、中序序列还原二叉树:

    Node getBinaryTree(int preOrder_arr[],int left,int right,int inOrder_arr[],int low,int high){
      //结束递归的条件
      if(left >= right){
        //如果只有一个节点,那么就结束递归
        return NULL;
      }
      int index,root,lcount = 0,rcount = 0;
      root = preOrder_arr[left];//有前序序列得到根节点
      index = getRoot(inOrder_arr,low,high,root);//在中序数组中获取根节点的下标
      //由根节点的下标,我们可以直到左子树有多少个节点,右子树有多少个节点
      lcount = index - low;
      rcount = high - index - 1;
      //创建根节点
      Node node = (Node)malloc(sizeof(struct NODE));
      node->val = root;
      //递归获得根节点的左子树
      node->left = getBinaryTree(preOrder_arr,left + 1,left + lcount + 1,inOrder_arr,low,index);
      //递归获得根节点的右子树
      node->right = getBinaryTree(preOrder_arr,left+lcount + 1,right,inOrder_arr,index + 1,high);
      return node;
    }
    

    中序序列、后序序列还原二叉树:

    //由中序序列、后序序列还原二叉树
    Node getBinaryTree2(int inOrder_arr[],int low,int high,int postOrder_arr[],int left,int right){
      if(left >= right){
        //如果只有一个节点,那么就结束递归
        return NULL;
      }
      int index,root,lcount = 0,rcount = 0;
      root = postOrder_arr[right - 1];//后序序列最后一个节点是根节点
      index = getRoot(inOrder_arr,low,high,root);//在中序序列中找到根节点的下标
      //创建根节点
      Node node = (Node)malloc(sizeof(struct NODE));
      node->val = root;
      //获取左右子数的节点个数
      lcount = index - low;
      rcount = high - index - 1;
     // printf("根节点的左子树有%d个,右子树有%d个\n",lcount,rcount);
      //创建按根节点的左子树
      node->left = getBinaryTree2(inOrder_arr,low,index,postOrder_arr,left,left + lcount);
      //创建根节点的右子树
      node->right = getBinaryTree2(inOrder_arr,index + 1,high,postOrder_arr,left + lcount,right - 1);
      return node;
    }
    

    测试运行代码:

    /*
    给出两种遍历序列(前序和中序、中序和后序),然后以这两种序列为依据还原二叉树
    1、根据前序序列、中序序列还原二叉树
    基本思路:
       1、定义两个数组,表示两种序列的输出
       2、由于前序序列,那么第一个数必定是一个根节点,所以我们有前序
       序列,在中序序列中找到根节点对应的下标,从而我们由中序序列也知道了
       根节点的左边是他的左子树,右边是他的右子树,那么我们将中序序列就划分成为了
       两个子数组,同时也有左、右子数的节点个数,将前序序列也划分成为2哥子数组
       3、重复步骤2,直到子数组中的只有一个节点或者没有,这时候结束递归
    
    2、根据中序序列、后序序列还原二叉树
    基本思路:和1的一样,只是在由后序序列找到根节点的值有所不同,因为后序序列的根节点
    在最后一个,其他的步骤相似
    
    请输入节点的个数:12
    请输入前序序列:10 9 7 6 8 15 14 11 14 19 18 21
    请输入中序序列:6 7 8 9 10 11 14 14 15 18 19 21
    请输入后序序列:6 8 7 9 11 14 14 18 21 19 15 10
    */
    #include<stdio.h>
    #include<stdlib.h>
    #define MAX_SIZE 100
    #define ERROR 0
    #define OK 1
    typedef struct NODE * Node;
    typedef Node * List;
    struct NODE{
       int val;
       Node left;
       Node right;
    };
    typedef struct QUEUE{
       List arr;
       int front;//队头指针
       int rear;//队尾指针
    }Queue;
    int init(Queue &s){
      s.arr = (List)malloc(sizeof(List) * MAX_SIZE);//定义一个指针类型的数组
      if(s.arr == NULL){
        return ERROR;
      }
    
      int i;
      //给叔组初始化之后还没有可以,还需要给所有的节点分配空间,如果没有这一步,那么就会发生报错
      for(i = 0; i < MAX_SIZE; i++){
        s.arr[i] = (Node)malloc(sizeof(struct NODE));
        if(s.arr[i] == NULL)
            return ERROR;
      }
      s.front = s.rear = 0;//将队头指针、队尾指针都初始为0
      return OK;
    }
    //压入队列
    int pushQueue(Queue &s,Node &node){
       if((s.rear + 1) % MAX_SIZE == s.front){
        //如果栈满了,返回ERROR
        printf("队列为满!!!\n");
        return ERROR;
       }
       s.arr[s.rear] = node;
       s.rear = (s.rear + 1) % MAX_SIZE;
       return OK;
    }
    int popQueue(Queue &s,Node &k){
       if(s.rear == s.front){
         //printf("队列为空!!!\n");
         return ERROR;
       }
       k = s.arr[s.front];
       s.front = (s.front + 1) % MAX_SIZE;
       return OK;
    }
    int getTop(Queue &s,Node &k){
       if(s.rear == s.front){
         //printf("队列为空!!!\n");
         return ERROR;
       }
       k = s.arr[s.front];
       return OK;
    }
    int isEmpty(Queue &s){
       return s.rear == s.front;//判断队列是否为空
    }
    int getSize(Queue &s){
       return (s.rear - s.front + MAX_SIZE)%MAX_SIZE;//获取队列的个数
    }
    //利用递归构建二叉树
    Node createBinaryTree(Node root,int x){
        if(root == NULL){
            root = (Node )malloc(sizeof(struct NODE));
            if(root == NULL){
                printf("创建节点失败!!!\n");
                exit(0);
            }
            root->val = x;
            root->left = NULL;
            root->right = NULL;
        }else{
           //如果根节点不为空,那么就绪要找打新节点的位置
           if(x < root->val){
            //如果新节点的值比当前节点的值小,那么就需要将其往当前节点的左子树方向找
              root->left = createBinaryTree(root->left,x);
           }else{
              root->right = createBinaryTree(root->right,x);
           }
        }
        return root;
    }
    //层序遍历
    void levelOrderTraversal2(Node root){
         Node t = root,k;
         Queue q;
         int size,i,count = 1;
         init(q);
         pushQueue(q,t);//将根节点压入队列中
         while(!isEmpty(q)){
            size = getSize(q);
            for(i = 1; i <= size; i++){
                popQueue(q,k);
                printf("%5d",k->val);
                //每跳出一个节点,那么就将它的左右子节点压入到队列中
                if(k->left != NULL){
                    pushQueue(q,k->left);
                }
                if(k->right != NULL){
                    pushQueue(q,k->right);
                }
            }
            printf("\n");
         }
    }
    //通过循环找树中的最小值
    Node findMin(Node root){
       Node current = root;
       while(current->left != NULL){
         current = current->left;
       }
       return current;
    }
    //获取二叉搜索树的高度
    int getHeight(Node root){
        int hl = 0,hr = 0,max;//hl表示的使左子树的高度,hr表示的使右子树的高度
        if(root != NULL){
           //当前的节点不为空,获取左右子树的高度
           hl = getHeight(root->left);
           hr = getHeight(root->right);
           max = hl > hr ? hl : hr;
           return max + 1;//左右子数高度的最大值加1就是当前节点的高度
        }else return 0;//如果当前节点为空,那么它的高度为0
    }
    /*
    查找值为x的节点,然后将其返回
    */
    Node findElement(Node root,int x){
       Node current = root;
       while(current != NULL){
         if(x < current->val)//如果当前的节点的值大于x的值,那么就往左子树的方向进行查找
            current = current->left;
         else if(x > current->val)
           current = current->right;
         else
              return current;
       }
       return NULL;//如果退出循环了,说明没有办法找到x的节点
    }
    /*
    删除值为x的节点(如果x出现了多次,那么就会删除第一个x)
    这时候我们需要将分为几种情况进行讨论:
      1、删除的节点是一个叶节点,直接将这个节点释放即可
      2、如果删除的节点含有一个子节点,那么这时候我们将这个删除节点的子节点
      替换掉这个节点即可
      3、如果这个删除节点含有两个子节点,那么我们将它的右子树中的最小节点的值赋给
      当前节点的值,那么这时候变成了删除右子树中的最小节点了(即前面的两种情况)
    */
    Node deleteElement(Node root,int x){
       if(root == NULL){
         printf("节点为空,无法进行删除操作!!!");
       }else if(x < root->val){
          root->left = deleteElement(root->left,x);
       }else if(x > root->val){
            root->right = deleteElement(root->right,x);
        }else{
          /*如果当前的节点是要删除的节点
          判断这个删除的节点是否为一个叶节点,如果是,那么直接将其变成NULL即可
          否则,如果这个删除节点只有一个子节点,那么就将子节点的值赋值给这个删
          除节点,然后将它的子节点变成为NULL,否则,如果这个删除节点含有两个子节点,那么
          就将遍历它的右子树,获取右子树中的最小值,然后将这个右子树的最小值赋值给这个
          删除节点的值,在将这个最小值变成NULL
          */
              if(root->left != NULL && root->right != NULL){
                //删除节点含有两个子节点
                Node tmp = findMin(root->right);
                root->val = tmp->val;
                root->right = deleteElement(root->right,tmp->val);
              }else{
                 /*
                 下面的代码如果使这样写的话,会发生错误的,为什么会这样呢?
                 其实很简单,因为这里已经包括了两种情况了,删除的节点是一个叶
                 节点或者只有一个子节点的节点,如果是这样写的话,并没有解决删
                 除节点是一个叶节点的情况,只是把这个删除节点的内存空间释放了
                   Node *t = root;
                 if(root->left != NULL){
                    root = root->left;
                 }else if(root->right != NULL){
                    root = root->right;
                 }
                 free(t);//释放删除的节点
                 */
                 Node t = root;
                 if(root->left == NULL){
                 /*
                 如果当前节点的左子节点为空,那么就用它的右子节点替换当前节
                 点,否则用左子节替换,这样进行判断的好处就是,如果这个删除节点
                 是一个叶节点,那么两个子节点都是空的,那么这时候root = root-
                 >right = NULL了,如果这个删除节点含有一个子节点,并且它的左
                 子节点为空,那么这个节点就用它的右子节点替换,下面的if判断同
                 理
                 */
                    root = root->right;
                 }else if(root->right == NULL){
                    root = root->left;
                 }
                 free(t);//释放删除的节点
    
              }
          }
       return root;
    }
    //利用递归的方式实现后序遍历
    void postOrderDisplay(Node root){
       if(root != 0){
          postOrderDisplay(root->left);
          postOrderDisplay(root->right);
          printf("%d ",root->val);
       }
    }
    //利用递归的方式实现前序遍历
    void preOrderDisplay(Node root){
       if(root != 0){
          printf("%d ",root->val);
          preOrderDisplay(root->left);
          preOrderDisplay(root->right);
       }
    }
    void input(int arr[],int n){
      int i;
      for(i = 0; i < n; i++)
        scanf("%d",&arr[i]);
    }
    int getRoot(int inOrder_arr[],int low,int high,int x){
       int i;
       for(i = low; i < high; i++){
        if(inOrder_arr[i] == x)
            return i;
       }
       return -1;
    }
    Node getBinaryTree(int preOrder_arr[],int left,int right,int inOrder_arr[],int low,int high){
      //结束递归的条件
      if(left >= right){
        //如果只有一个节点,那么就结束递归
        return NULL;
      }
      int index,root,lcount = 0,rcount = 0;
      root = preOrder_arr[left];//有前序序列得到根节点
      index = getRoot(inOrder_arr,low,high,root);//在中序数组中获取根节点的下标
      //由根节点的下标,我们可以直到左子树有多少个节点,右子树有多少个节点
      lcount = index - low;
      rcount = high - index - 1;
      //创建根节点
      Node node = (Node)malloc(sizeof(struct NODE));
      node->val = root;
      //递归获得根节点的左子树
      node->left = getBinaryTree(preOrder_arr,left + 1,left + lcount + 1,inOrder_arr,low,index);
      //递归获得根节点的右子树
      node->right = getBinaryTree(preOrder_arr,left+lcount + 1,right,inOrder_arr,index + 1,high);
      return node;
    }
    //由中序序列、后序序列还原二叉树
    Node getBinaryTree2(int inOrder_arr[],int low,int high,int postOrder_arr[],int left,int right){
      if(left >= right){
        //如果只有一个节点,那么就结束递归
        return NULL;
      }
      int index,root,lcount = 0,rcount = 0;
      root = postOrder_arr[right - 1];//后序序列最后一个节点是根节点
      index = getRoot(inOrder_arr,low,high,root);//在中序序列中找到根节点的下标
      //创建根节点
      Node node = (Node)malloc(sizeof(struct NODE));
      node->val = root;
      //获取左右子数的节点个数
      lcount = index - low;
      rcount = high - index - 1;
     // printf("根节点的左子树有%d个,右子树有%d个\n",lcount,rcount);
      //创建按根节点的左子树
      node->left = getBinaryTree2(inOrder_arr,low,index,postOrder_arr,left,left + lcount);
      //创建根节点的右子树
      node->right = getBinaryTree2(inOrder_arr,index + 1,high,postOrder_arr,left + lcount,right - 1);
      return node;
    }
    int main(){
      int preOrder_arr[MAX_SIZE],inOrder_arr[MAX_SIZE],postOrder_arr[MAX_SIZE];//定义两个数组,分别表示前序序列、中序序列
      int n,i;
      Node root;
      printf("请输入节点的个数:");
      scanf("%d",&n);
      printf("请输入前序序列:");
      input(preOrder_arr,n);
      printf("请输入中序序列:");
      input(inOrder_arr,n);
      printf("请输入后序序列:");
      input(postOrder_arr,n);
      root = getBinaryTree(preOrder_arr,0,n,inOrder_arr,0,n);
      printf("递归实现由前序序列、中序序列还原的二叉树的后序遍历:");
      postOrderDisplay(root);
      printf("\n");
      root = getBinaryTree2(inOrder_arr,0,n,postOrder_arr,0,n);
      printf("递归实现由中序序列、后序序列还原的二叉树的前序遍历:");
      preOrderDisplay(root);
      printf("\n两种序列还原的二叉树的高度为:");
      printf("%d\n",getHeight(root));
      printf("请输入要删除的节点:");
      while(scanf("%d",&n) != EOF){
          if(n == 0)
            break;
          root = deleteElement(root,n);
          printf("删除节点之后二叉树的后序遍历:");
          postOrderDisplay(root);
          printf("\n删除节点之后的二叉树的高度为:");
          printf("%d\n",getHeight(root));
          printf("删除节点之后的层序遍历:\n");
          levelOrderTraversal2(root);
          printf("请输入要删除的节点:");
      }
      return 0;
    }
    
    
    
    下一篇:没有了