从C中的BST删除节点

时间:2016-10-01 11:27:40

标签: c binary-search-tree nodes

我试图理解这个在线创建的功能,用于删除BST中的节点。有些事情我无法理解

这是代码:

struct Node* Delete(struct Node *root, int data) {
  if (root == NULL) {
     return NULL;
  }
  if (data > root->data) {  // data is in the left sub tree.
      root->left = Delete(root->left, data);
  } else if (data > root->data) { // data is in the right sub tree.
      root->right = Delete(root->right, data);
  } else {
     // case 1: no children
     if (root->left == NULL && root->right == NULL) {
        delete(root); // wipe out the memory, in C, use free function
        root = NULL;
     }
     // case 2: one child (right)
     else if (root->left == NULL) {
        struct Node *temp = root; // save current node as a backup
        root = root->right;
        delete temp;
     }
     // case 3: one child (left)
     else if (root->right == NULL) {
        struct Node *temp = root; // save current node as a backup
        root = root->left;
        delete temp;
     }
     // case 4: two children
     else {
        struct Node *temp = FindMin(root->right); // find minimal value of right sub tree
        root->data = temp->data; // duplicate the node
        root->right = Delete(root->right, temp->data); // delete the duplicate node
     }
  }
  return root; // parent node can update reference
}

问题:

1)为什么

if (data > root->data) {  // data is in the left sub tree.
          root->left = Delete(root->left, data);

不应该是if(data < root->data)吗? (紧随其后两行代码相同)

2)函数返回指向节点的指针,这是否意味着在main函数中我必须做这样的事情?

int main(){
struct Node *tree=malloc(sizeof(Node));
...
struct Node *new_tree=malloc(sizeof(Node));
new_tree= Delete(tree,24);

所以函数用没有带有val 24的节点的新树替换旧树?如果我想要函数为void类型我应该使用双指针吗?

1 个答案:

答案 0 :(得分:2)

对于你的第一个问题,你应该是:if(data < root->data)。 对于第二个问题并不完全正确。你显然应该定义一个指针头,它是树的头部,并创建一个向bst插入数据的函数,所以这个函数执行malloc。您在main中的所有内容都是头部指针在开始时初始化为NULL,因此它应该如下所示:

int main(){
struct Node *tree=NULL;
int number=...; 
...
input_to_bst(&tree,number);
...
new_tree= Delete(tree,24);

另请注意,新树不需要malloc,因为您的函数返回已经显示给结构的指针,您所做的是new_tree也将指向此结构。

对于你的最后一个问题,当然你可以传递双指针(实际上我在input_to_bst(&tree);的定义中遵循这种方式。)

函数input_to_bst定义的示例可以是:

void input_to_bst(treeptr* head,int number){
  if((*head)==NULL){
       (*head)=(treeptr)malloc(sizeof(struct tree));
       (*head)->data=number;
       (*head)->left=NULL;
       (*head)->right=NULL;
  }
  else{
      if(((*head)->data)>number) input_to_bst(&((*head)->left),number);
      else (((*head)->data)<number) input_to_bst(&((*head)->right),number);
  }
}

我们假设我们已经定义了结构:

struct tree{
    int data;
    struct tree* right;
    struct tree* left;
};

typedef struct tree* treeptr;