Question

我正在制作一个程序来制作一个二进制搜索树，它从用户那里获取输入。我在下面的代码中故意展示了两个搜索函数。

问题：搜索功能2 正常工作但搜索功能1 无法正常工作（每次评论其他用户后使用）。为什么会这样？

我尝试进行干运行并构建递归堆栈，这对我来说很好。但不知何故，我认为搜索功能1 无法在链表中进行链接以插入元素。这就是我尝试inorder traversal

时无法获得正确输出的原因

任何帮助都将非常感谢！

#include <stdio.h>
#include <stdlib.h>

struct node{
    int data;
    struct node *left;
    struct node *right;
};
struct node *root = NULL;

struct node *newNode(int data){
    struct node *temp = (struct node *)malloc(sizeof(struct node));
    temp->data = data;
    temp->left = NULL;
    temp->right = NULL;
    return temp;
}

//SEARCH FUNCTION 1: this does not work correctly
void search(struct node *t,int data){
    if(t){
        if(data > t->data){
            search(t->right,data);
        }else{
            search(t->left,data);
        }
    }else{
        t = newNode(data);
    }
}

//SEARCH FUNCTION 2: this works fine and inserts the element correctly
void search(struct node *t, int data){
    if(data < t->data && t->left != NULL){
        search(t->left, data);
    }else if(data < t->data && t->left == NULL){
        t->left = newNode(data);
    }else if(data > t->data && t->right != NULL){
        search(t->right,data);
    }else{
        t->right = newNode(data);
    }
}

void insertNode(int data){
    if(!root){
        root = newNode(data);
        return;
    }
    search(root, data);
}

void inorder(struct node *t){
    if(t){
        if(t->left){
            inorder(t->left);   
        }
        printf("%d ->", t->data);
        if(t->right){
            inorder(t->right);  
        }
    }
}

int main(){
    int step, data;
    while(1){
        printf("1. Insert element\n");
        printf("2. Print tree\n");
        scanf("%d",&step);
        switch(step){
            case 1: printf("enter element to be inserted\n");
                scanf("%d",&data);
                insertNode(data);
                break;
            case 2:inorder(root); 
                printf("\n");
                break;
        }
    }
    return 0;
}

Answer 1

问题是语句t = newNode(data)分配给局部变量，因此从函数返回后结果立即丢失。

在这种情况下，一个解决方案是双重间接，因为您不想修改指针指向的东西，而是指针本身。

void search(struct node **pt,int data)
{
    struct node *t = *pt;
    if (t) {
        if (data > t->data) {
            search(&t->right,data);
        } else {
            search(&t->left,data);
        }
    } else {
        *pt = newNode(data);
    }
}

Answer 2

第一次搜索功能：

void search(struct node *t,int data){
    ...
    t = newNode(data);
}

然后在第二个搜索功能中执行：

void search(struct node *t, int data){
    ...
    t->right = newNode(data);
}

会记住作业，而第一部则不会，因为当你要递归时，改变将会丢失。

在第二种情况下，您将分配newNode()返回到作为指针的结构的数据成员的内容，而结构在此函数中传递指针。但是，在第一种情况下，您将结果分配给仅由一个指针传递的结构。如果使用双指针，事情会有所不同。

使用C中的递归插入二叉搜索树

2 个答案: