在C中的循环上将节点插入BST

时间:2017-03-05 23:25:03

标签: c binary-search-tree user-input

我正在尝试插入我的BST,但我正在努力创建一个循环。 当我逐个插入时,代码可以工作,但是当我尝试将其放入循环时,它无法正确插入。

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <string.h> // for strcmp()
#include <ctype.h> // for toupper()

typedef struct BstNode{
  //char name[20];
  // int data;
  struct BstNode* left;
  struct BstNode* right;
  char* name;
}BstNode;

typedef int (*Compare)(const char*, const char*); // makes comparisons easier

/* Returns pointer address to newly created node */
BstNode* createNode(char* name){ 
  BstNode* newNode = (BstNode*)malloc(sizeof(BstNode)); // Allocates memory for the newNode
  newNode->name = name; // newNode->data is like newNode.data
  newNode->left= NULL;
  newNode->right = NULL;
  return newNode;
}

//insert node into Tree recursively
BstNode* insertNode(BstNode* node, char* name, Compare cmp){
  int i;
  /* char *s1 = node->name;
     char *s2 = name;
     printf("s1: %s, s2: %s\n", s1,s2);
     i = strcmp(s1, s2); // if =1, s1 is greater
     printf("i: %d\n", i); */

  if(node == NULL){// if tree is empty
    // printf("inside NULL\n");
    node = createNode(name);
    //return node;
  }
  else{
    i = cmp (name, node->name); // sweet
    if(i == -1){
      // printf("inside left\n");
      node->left = insertNode(node->left, name, cmp);
      //return node;
    }
    else if(i == 1){
      // printf("inside right\n");
      node->right =  insertNode(node->right, name, cmp);
      //return node;
    }
    else if(i == 0 ){ //avoid duplicates for now
      // printf("inside 0\n");
      printf("Name is in BST\n");
      return NULL;
    }
  }
  return node;
}

BstNode* printTree(BstNode* node){
  if(node == NULL){
    return NULL;
  }
  printTree(node->left);
  printf("%s\n",node->name);
  printTree(node->right);
}

int CmpStr(const char* a, const char* b){
  return (strcmp (a, b));     // string comparison instead of pointer comparison
}
//void Insert(Person *root, char name[20]);

int main(){
  BstNode* root = NULL; // pointer to the root of the tree
  char buf[100];
  char option = 'a';

  while(1) { 
    printf("Enter employee name");
    scanf("%s",buf); 
    printf ("Inserting %s\n", buf);
    root =  insertNode(root, buf, (Compare)CmpStr);
    printTree(root);
  }
}

我可以做root = insertNode(root,name,(Compare)CmpStr) 在代码中多次,但如果我尝试用用户输入循环它,它将无法正确插入。我不确定是否与我使用scanf()或root设置不正确的事实有关。我也尝试过使用fgets(),但我不太清楚如何使用它并继续弄乱它。 任何帮助表示赞赏。

1 个答案:

答案 0 :(得分:0)

在循环中,总是将相同的缓冲区传递给insert函数;您的createNode不会复制缓冲区的内容,而是存储对(始终)相同缓冲区的引用;因此,在插入后更改缓冲区内容也会改变内容&#34;之前插入的节点。

我建议将newNode->name = name中的createNode替换为newNode->name = strdup(name)。这实际上将复制传递的&#34;内容&#34;并让你的BST控制要保留的内存。因此,以后在删除节点时不要忘记释放此内存。