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

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()情况下看到，将newNode()返回的值分配给作为指针的结构的数据成员，而在此函数中向该结构传递了一个指针。 但是，在第一种情况下，您将结果分配给仅由一个指针传递的结构。 如果使用双指针，情况将有所不同。