用C语言写一段代码,实现二叉树节点的删除操作,具体过程为根据输入的节点,在二叉树中找到该节点,删除该节点的左子树
时间: 2023-09-25 13:08:10 浏览: 235
以下是用C语言实现二叉树节点删除操作的示例代码:
```c
#include <stdio.h>
#include <stdlib.h>
typedef struct Node {
int data;
struct Node *left;
struct Node *right;
} Node;
Node* createNode(int data) {
Node* newNode = (Node*) malloc(sizeof(Node));
newNode->data = data;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
Node* insert(Node* root, int data) {
if (root == NULL) {
root = createNode(data);
} else if (data <= root->data) {
root->left = insert(root->left, data);
} else {
root->right = insert(root->right, data);
}
return root;
}
Node* find(Node* root, int data) {
if (root == NULL) {
return NULL;
} else if (root->data == data) {
return root;
} else if (data < root->data) {
return find(root->left, data);
} else {
return find(root->right, data);
}
}
Node* delete(Node* root, int data) {
if (root == NULL) {
return NULL;
} else if (data < root->data) {
root->left = delete(root->left, data);
} else if (data > root->data) {
root->right = delete(root->right, data);
} else {
// Case 1: No Child
if (root->left == NULL && root->right == NULL) {
free(root);
root = NULL;
}
// Case 2: One Child
else if (root->left == NULL) {
Node* temp = root;
root = root->right;
free(temp);
} else if (root->right == NULL) {
Node* temp = root;
root = root->left;
free(temp);
}
// Case 3: Two Children
else {
Node* temp = root->right;
while (temp->left != NULL) {
temp = temp->left;
}
root->data = temp->data;
root->right = delete(root->right, temp->data);
}
}
return root;
}
void inorderTraversal(Node* root) {
if (root == NULL) {
return;
}
inorderTraversal(root->left);
printf("%d ", root->data);
inorderTraversal(root->right);
}
int main() {
Node* root = NULL;
root = insert(root, 50);
insert(root, 30);
insert(root, 20);
insert(root, 40);
insert(root, 70);
insert(root, 60);
insert(root, 80);
printf("Before deletion: ");
inorderTraversal(root);
int dataToDelete = 20;
Node* nodeToDelete = find(root, dataToDelete);
if (nodeToDelete == NULL) {
printf("\n%d not found in the tree\n", dataToDelete);
} else {
delete(root, dataToDelete);
printf("\nAfter deletion of %d: ", dataToDelete);
inorderTraversal(root);
printf("\n");
}
return 0;
}
```
在上面的代码中,我们首先定义了一个 `Node` 结构体来表示二叉树的节点,然后实现了三个函数:
- `createNode()`:创建一个新的节点并返回指向该节点的指针;
- `insert()`:向二叉树中插入一个新的节点;
- `find()`:在二叉树中查找指定的节点。
最重要的是 `delete()` 函数,它实现了删除二叉树节点的逻辑。这个函数使用递归的方式在二叉树中查找要删除的节点,并根据节点的情况执行三种不同的删除操作:
- Case 1: No Child:如果要删除的节点没有子节点,直接删除该节点;
- Case 2: One Child:如果要删除的节点只有一个子节点,将该节点的父节点指向该节点的子节点,并删除该节点;
- Case 3: Two Children:如果要删除的节点有两个子节点,找到右子树中最小的节点,将该节点的值赋给要删除的节点,并递归地删除右子树中的该节点。
最后,我们在 `main()` 函数中创建一个二叉树并删除其中的一个节点,然后输出删除节点后的二叉树中序遍历的结果。
阅读全文