二叉树删除——看不懂一些指针

Question

我对这个二叉树删除有疑问。代码是

#include<stdio.h>
#include<stdlib.h>
struct node
{
    int key;
    struct node *left, *right;
};
struct node *newNode(int item)
{
    struct node *temp =  (struct node *)malloc(sizeof(struct node));
    temp->key = item;
    temp->left = temp->right = NULL;
    return temp;
}
void inorder(struct node *root)
{
    if (root != NULL)
    {
        inorder(root->left);
        printf("%d ", root->key);
        inorder(root->right);
    }
}
struct node* insert(struct node* node, int key)
{
    if (node == NULL) return newNode(key);

    if (key < node->key)
        node->left  = insert(node->left, key);
    else
        node->right = insert(node->right, key);
    return node;
}
struct node * minValueNode(struct node* node)
{
    struct node* current = node;
     while (current->left != NULL)
        current = current->left;

    return current;
}
struct node* deleteNode(struct node* root, int key)
{
   if (root == NULL) return root;

   if (key < root->key)
        root->left = deleteNode(root->left, key);
     else if (key > root->key)
        root->right = deleteNode(root->right, key);
   else
    {
        if (root->left == NULL)
        {
            struct node *temp = root->right;
            free(root);
            return temp;
        }
        else if (root->right == NULL)
        {
            struct node *temp = root->left;
            free(root);
            return temp;
        }

    struct node* temp = minValueNode(root->right);

         root->key = temp->key;

       root->right = deleteNode(root->right, temp->key);
    }
    return root;
}

int main()
{
    struct node *root = NULL;
    root = insert(root, 50);
    root = insert(root, 30);
    root = insert(root, 20);
    root = insert(root, 40);
    root = insert(root, 70);
    root = insert(root, 60);
    root = insert(root, 80);

    printf("Inorder traversal of the given tree \n");
    inorder(root);

    printf("\nDelete 20\n");
    root = deleteNode(root, 20);
    printf("Inorder traversal of the modified tree \n");
    inorder(root);

    printf("\nDelete 30\n");
    root = deleteNode(root, 30);
    printf("Inorder traversal of the modified tree \n");
    inorder(root);

    printf("\nDelete 50\n");
    root = deleteNode(root, 50);
    printf("Inorder traversal of the modified tree \n");
    inorder(root);

    return 0;
}

我看不懂的台词是：

if (root->left == NULL)
        {
            struct node *temp = root->right;
            free(root);
            return temp;
        }
        else if (root->right == NULL)
        {
            struct node *temp = root->left;
            free(root);
            return temp;
        }

此代码实际上是否返回 NULL 值，那么我可以在这两种情况下将代码重写为 struct node *temp=NULL 但是当我这样做时不会显示最后一个中序值。

Answer 1

您查询的代码是处理要删除的节点是树的根节点的代码的一部分。它检查根节点是否只有一个子节点（或零个），在这种情况下，只需将一个子节点作为树的新根即可执行删除。

特别注意，当root->left == NULL 时，选择的是右子节点，而不是左子节点（并临时记录在temp 中），反之亦然。无论哪种方式，原始根节点都会被释放，因为它不再在树中，并且无法通过函数即将返回的新根指针访问。

仅当根节点最初是树中的唯一节点时，该函数才返回NULL。