这是我创建和插入二叉搜索树的代码
struct BTNode
{
int info;
struct BTNode *left, *right;
};
struct BTNode* create(int data)
{
struct BTNode *root;
struct BTNode *n = malloc(sizeof(struct BTNode));
n->info = data;
n->left = NULL;
n->right = NULL;
root = n;
return(root);
}
struct BTNode* insert(struct BTNode *root, int data)
{
struct BTNode *ptr;
struct BTNode *n = malloc(sizeof(struct BTNode));
if (n == NULL)
printf("nOUT OF MEMORY!n");
n->info = data;
n->left = NULL;
n->right = NULL;
ptr = root;
while (ptr != NULL){
if (data < ptr->info){
if (ptr->left == NULL)
ptr->left = n;
ptr = ptr->left;
}
else if (data > ptr->info){
if (ptr->right == NULL)
ptr->right = n;
ptr = ptr->right;
}
}
return(n);
}
这是main((函数
int main()
{
struct BTNode *root = NULL;
int choice, data;
printf("nWrite the root data: ");
scanf("%d", &data);
root = create(data);
while (1){
printf("n1.Insert 2.Preorder 3.Exitn");
scanf("%d", &choice);
switch(choice){
case 1:
printf("nWrite the data: ");
scanf("%d", &data);
insert(root, data);
break;
我能够创建根节点,但每当我尝试插入数据时,我都会提供我的数据,编译器会停止执行任何操作。知道为什么会这样吗?
你的while()
循环会永远持续下去,因为即使你找到插入节点的地方,你也会继续前进:
while(ptr!=NULL){
if(data<ptr->info){
if(ptr->left==NULL)
ptr->left=n;
ptr=ptr->left;
}
else if(data>ptr->info){
if(ptr->right==NULL)
ptr->right=n;
ptr=ptr->right;
}
}
插入节点后需要断开while()
循环:
while (ptr != NULL) {
if (data < ptr->info) {
if (ptr->left == NULL) {
ptr->left = n;
break;
}
ptr = ptr->left;
} else if (data > ptr->info) {
if (ptr->right == NULL) {
ptr->right = n;
break;
}
ptr = ptr->right;
}
}
此外,检查malloc()
是否失败的奖励积分
struct BTNode *n = malloc(sizeof(struct BTNode));
if (n == NULL)
printf("nOUT OF MEMORY!n");
但是无论如何简单地继续的缺点,如果失败malloc()
您应该退出该功能
struct BTNode *n = malloc(sizeof(struct BTNode));
if (n == NULL) {
printf("nOUT OF MEMORY!n");
return NULL:
}
然后,当然,调用insert()
的代码应该知道如果insert()
返回 NULL 该怎么办。