欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

c/c++ 二叉排序树

程序员文章站 2022-08-05 09:17:50
c/c++ 二叉排序树 概念: 左树的所有节点的值(包括子节点)必须小于中心节点,右树所有节点的值(包括子节点)必须大于中心节点。 不允许有值相同的节点。 二叉排序树的特点: 中序遍历后,就是从小到大排序了。 根节点的最左边的值,就是树中最小的值。 根节点的最右边的值,就是树中最大的值。 创建二叉排 ......

c/c++ 二叉排序树

概念:

左树的所有节点的值(包括子节点)必须小于中心节点,右树所有节点的值(包括子节点)必须大于中心节点。

不允许有值相同的节点。

c/c++ 二叉排序树

二叉排序树的特点:

  • 中序遍历后,就是从小到大排序了。
  • 根节点的最左边的值,就是树中最小的值。
  • 根节点的最右边的值,就是树中最大的值。

创建二叉排序树的思路:

  • 用递归的方式
  • 和根节点比较大小
  • 比根节点小的话,用递归去和根节点的左节点比较,至到找到合适的位置
  • 比根节点大的话,用递归去和根节点的右节点比较,至到找到合适的位置

二叉排序树的一些实用函数

init_bst 初始化二叉排序树
insert_bst_tree 插入树的节点
min 求树中最小节点
max 求树中最大节点
sort 排序二叉树(中序遍历就是从小到大排序了)
remove_bst 删除节点

删除节点

pattern1:要被删除的节点是root节点

  • 方案1:用根节点左树中的最大的节点作为新的根节点

删除45
c/c++ 二叉排序树

  • 方案2:用根节点又树中的最小的节点作为新的根节点

删除45
c/c++ 二叉排序树

pattern2:要被删除的节点是其父节点的左树,并且要被删除的节点有右树

删除12
c/c++ 二叉排序树

pattern3:要被删除的节点是其父节点的左树,并且要被删除的节点无右树

删除12
c/c++ 二叉排序树

pattern4:要被删除的节点是其父节点的右树,并且要被删除的节点无左树

删除53
c/c++ 二叉排序树

pattern5:要被删除的节点是其父节点的右树,并且要被删除的节点有左树

删除100
c/c++ 二叉排序树

bst.h

#ifndef __BST__
#define __BST__

#include <stdio.h>
#include <malloc.h>
#include <assert.h>

#define T int
#define FALSE 0
#define TRUE 1
#define BOOL int

typedef struct BSTNode{
  T data;
  struct BSTNode* left;
  struct BSTNode* right;
}BSTNode;

typedef struct BST{
  BSTNode* root;
}BST;

//初始化二叉排序树
void init_bst(BST* bst);
//插入树的节点
BOOL insert_bst_node(BSTNode** t, T x);
BOOL insert_bst_tree(BST* bst, T x);
//求树中最小节点
T min(BST* bst);
//求树中最大节点
T max(BST* bst);
//排序
void sort(BST* bst);
//查找父节点
BSTNode* get_parent(BST* bst, BSTNode* tar);
//删除节点
BOOL remove_bst(BST* bst, T key);
//搜索节点
BSTNode* search_bst(BST* bst, T key);
//搜索节点
BSTNode* search_bst1(BST* bst, T key);
//清空树
void clear_bst(BST* bst);

#endif

bst.c

#include "bst.h"

//初始化二叉排序树
void init_bst(BST* bst){
  bst->root = NULL;
}

//插入树的节点
BOOL insert_bst_node(BSTNode** t, T x){
  if(*t == NULL){
    *t = (BSTNode*)malloc(sizeof(BSTNode));
    assert(NULL != *t);
    (*t)->data  = x;
    (*t)->left  = NULL;
    (*t)->right = NULL;
    return TRUE;
  }
  else if(x < (*t)->data){
    insert_bst_node(&((*t)->left), x);
  }
  else if(x > (*t)->data){
    insert_bst_node(&((*t)->right), x);
  }
  return FALSE;
}
BOOL insert_bst_tree(BST* bst, T x){
  return insert_bst_node(&(bst->root), x);
}

//求树中最小节点
T min_node(BSTNode* t){
  while(t->left != NULL)
    t = t->left;
  return t->data;
}
T min(BST* bst){
  assert(bst->root != NULL);
  return min_node(bst->root);
}
//求树中最大节点
T max_node(BSTNode* t){
  while(t->right != NULL){
    t = t->right;
  }
  return t->data;
}
T max(BST* bst){
  assert(bst->root != NULL);
  return max_node(bst->root);
}
//二叉树中序排序
void sort_node(BSTNode* t){
  if(NULL == t){
    return;
  }else{
    sort_node(t->left);
    printf("%d ", t->data);
    sort_node(t->right);
  }
}
void sort(BST* bst){
  assert(NULL != bst->root);
  sort_node(bst->root);
}
//搜索节点
BSTNode* search_node(BSTNode* t, T key){
  if(NULL == t || t->data == key){
    return t;
  }
  else{
    BSTNode* p;
    p = search_node(t->left, key);
    if(NULL == p){
      p = search_node(t->right, key);
    }
    return p;
  }
}
BSTNode* search_bst(BST* bst, T key){
  return search_node(bst->root, key);
}
BSTNode* search_node1(BSTNode* t, T key){
  if(NULL == t || t->data == key){
    return t;
  }
  else{
    if(key < t->data){
      search_node1(t->left, key);
    }
    else{
      search_node1(t->right, key);
    }
  }
}
BSTNode* search_bst1(BST* bst, T key){
  return search_node1(bst->root, key);
}
//清空树
void clear_node(BSTNode** t){
  if(NULL != *t){
    clear_node(&((*t)->left));
    clear_node(&((*t)->right));
    free(*t);
    *t = NULL;
  }
}
void clear_bst(BST* bst){
  clear_node(&bst->root);
}

//查找父节点
BSTNode* get_parent_node(BSTNode* t, BSTNode* tar){
  if(NULL == t || NULL == tar)return NULL;
  if(t->left == tar || t->right == tar){
    return t;
  }
  else{
    BSTNode* p = NULL;
    p = get_parent_node(t->left, tar);
    if(NULL == p){
      p = get_parent_node(t->right, tar);
    }
    return p;
  }
}
BSTNode* get_parent(BST* bst, BSTNode* tar){
  return get_parent_node(bst->root, tar);
}
BOOL remove_bst(BST* bst, T key){

  BSTNode* tar = search_bst(bst, key);
  //树为空或者要删除的节点不存在,返回失败
  if(bst->root == NULL || NULL == tar) return FALSE;

  BSTNode* parent = get_parent(bst, tar);
  //因为要被删除的顶点有左子节点,所以要找到以左子节点为根的右子节点中值最大的
  BSTNode* X = NULL;
  if(NULL != tar->left){
    X = tar->left;
    while(X->right != NULL){
      X = X->right;
    }
      
    //因为要被删除的顶点的左子节点,有右子节点,所以要找到最大的
    if(X != tar->left){
      //找到最大节点的父节点
      BSTNode* X1 = get_parent(bst, X);
      //最大节点的父节点的右边指向最大节点的左边
      X1->right = X->left;
    
      //最大节点的左边代替被删除节点的左边,右边代替右边
      X->left = tar->left;
      X->right = tar->right;
    }
    //因为要被删除的顶点的左子节点,没有右子节点,所以它就是最大的
    else{
      X->right = tar->right;
    }
  }
  //因为要被删除的顶点没有左子节点,所以要找到以右子节点为根的左子节点中值最小的
  else{
    X = tar->right;
    //要被删除的节点既没有左节点,也没有右节点
    if(NULL == X){
      //找到父节点
      BSTNode* X2 = get_parent(bst, X);
      //要被删除的节点不是根节点
      if(parent != NULL){
    //要被删除的顶点在父节点的左边
    if(tar->data < parent->data){
      parent->left = X;
    }
    //要被删除的顶点在父节点的右边
    else{
      parent->right = X;
    }
      }
      else{
    bst->root = NULL;
      }
      free(tar);
      return TRUE;
    }
    while(X->left != NULL){
      X = X->left;
    }
    //因为要被删除的顶点的右子节点,有左子节点,所以要找到最小的
    if(X != tar->right){
      //找到最小节点的父节点
      BSTNode* X1 = get_parent(bst, X);
      //最小节点的父节点的左边指向最小节点的右边
      X1->left = X->right;

      //最小节点的左边代替被删除节点的左边,右边代替右边
      X->right = tar->right;
      X->left = tar->left;
    }
  }
  //要被删除的节点不是根节点
  if(parent != NULL){
    //要被删除的顶点在父节点的左边
    if(tar->data < parent->data){
      parent->left = X;
    }
    //要被删除的顶点在父节点的右边
    else{
      parent->right = X;
    }
  }
  else{
    bst->root = X;
  }
  free(tar);
}

bstmain.c

#include "bst.h"

int main(){
  BST bst;
  init_bst(&bst);

  //patten1 目标节点是root,root没有右子节点,左子节点中有右子节点
  //T ar[] = {45,12,3,37,24,38};
  //patten2 目标节点是root,root没有右子节点,左子节点中没有右子节点
  //T ar[] = {45,12,3};
  //patten3 目标节点是root,只有root节点
  //T ar[] = {45};
  //patten4 目标节点是root,root有右子节点,右子节点中没有左子节点
  //T ar[] = {45,12,53,3,37,100,24};
  //patten5 目标节点是root,root有右子节点,右子节点中有左子节点
  //T ar[] = {45,12,53,3,37,100,24,61,90,78};

  //patten6 目标节点(8)不是root,目标节点有左子节点,左子节点没有右边
  //T ar[] = {45,12,53,3,27,2,4,24,1,6,5,8,7};
  //patten7 目标节点(12)不是root,目标节点有左子节点,左子节点有右边
  //T ar[] = {45,12,53,3,27,2,4,24,1,6,5,8,7};
  //patten8 目标节点(120)不是root,目标节点没有左子节点,右子节点没有左边
  T ar[] = {45,12,53,3,37,52,100,2,4,24,51,61,120,1,6,90,130,5,8,78,126,140,7,124,127,125};

  //T ar[] = {45,12,53,3,37,100,24,61,90,78};
  //T ar[] = {45,3,4,12,53};
  int n = sizeof(ar) / sizeof(T);
  for(int i = 0; i < n; ++i){
    insert_bst_tree(&bst, ar[i]);
  }

  sort(&bst);
  printf("\n");

  //删除节点
  remove_bst(&bst, 45);
  sort(&bst);
  printf("\n");
  
  clear_bst(&bst);
}

编译方法:gcc -g bst.c bstmain.c