我正在尝试插入我的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(),但我不太清楚如何使用它并继续弄乱它。 任何帮助表示赞赏。
答案 0 :(得分:0)
在循环中,总是将相同的缓冲区传递给insert函数;您的createNode
不会复制缓冲区的内容,而是存储对(始终)相同缓冲区的引用;因此,在插入后更改缓冲区内容也会改变内容&#34;之前插入的节点。
我建议将newNode->name = name
中的createNode
替换为newNode->name = strdup(name)
。这实际上将复制传递的&#34;内容&#34;并让你的BST控制要保留的内存。因此,以后在删除节点时不要忘记释放此内存。