c - 在C中将元素插入BST

Question

我有一个如下所示的结构类型:

typedef struct TreeNode{
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
} TreeNode;

将元素添加到我的

TreeNode* insert(TreeNode *root, int val){
TreeNode* a;
    a = root;
    int i;
    if(a==NULL){
        a -> val = val;
        a -> left = NULL;
        a -> right = NULL;
    }
    else if(a->val < val){
        return insert(root->left,val);
    }
    else if(a->val > val)
        return insert(root->right,val);
}

当我评估它时，它没有给出任何输出。我有什么错？

Answer 1

1. 在a == NULL情况下，需要为节点分配内存。否则 a -> val 是非法的。
2. 还需要在代码中添加一个返回。您可以返回(a)。然后，当您调用该函数时，将其调用为 root = insert(root, val)

代码如下。

TreeNode* insert(TreeNode *root, int val){
  TreeNode* a;
  a = root;
  int i;
  if(a==NULL){
    // Allocate memory here
    a = malloc(sizeof (root));
    if (a== NULL)
    {
       // Malloc error, You can exit the program or print a debug message here
    }

    a -> val = val;
    a -> left = NULL;
    a -> right = NULL;
  }
  else if(a->val < val){
    return insert(root->left,val);
  }
  else if(a->val > val)
    return insert(root->right,val);
  return a;
}