下面是二叉树的基本操作代码,包括创建二叉树、插入节点、删除节点、查找节点和遍历等操作。每个函数都有详细的注释解释其功能和实现方法。
```c
#include <stdio.h>
#include <stdlib.h>
// 二叉树节点结构
typedef struct TreeNode {
int data; // 节点数据
struct TreeNode* left; // 左子节点指针
struct TreeNode* right; // 右子节点指针
} TreeNode;
// 创建新节点
TreeNode* createNode(int data) {
TreeNode* newNode = (TreeNode*)malloc(sizeof(TreeNode));
if (newNode == NULL) {
printf("内存分配失败\n");
return NULL;
}
newNode->data = data;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
// 插入节点
TreeNode* insertNode(TreeNode* root, int data) {
if (root == NULL) {
// 如果树为空,创建一个新节点作为根节点
root = createNode(data);
} else {
if (data <= root->data) {
// 如果插入值小于等于当前节点值,递归插入左子树
root->left = insertNode(root->left, data);
} else {
// 如果插入值大于当前节点值,递归插入右子树
root->right = insertNode(root->right, data);
}
}
return root;
}
// 删除节点
TreeNode* deleteNode(TreeNode* root, int data) {
if (root == NULL) {
return root;
} else if (data < root->data) {
// 递归删除左子树中的节点
root->left = deleteNode(root->left, data);
} else if (data > root->data) {
// 递归删除右子树中的节点
root->right = deleteNode(root->right, data);
} else {
// 找到要删除的节点
if (root->left == NULL && root->right == NULL) {
// 情况1: 要删除的节点为叶子节点
free(root);
root = NULL;
} else if (root->left == NULL) {
// 情况2: 要删除的节点只有右子节点
TreeNode* temp = root;
root = root->right;
free(temp);
} else if (root->right == NULL) {
// 情况3: 要删除的节点只有左子节点
TreeNode* temp = root;
root = root->left;
free(temp);
} else {
// 情况4: 要删除的节点有左右子节点
// 找到右子树中的最小节点作为替代节点
TreeNode* temp = findMin(root->right);
root->data = temp->data;
// 递归删除右子树中的替代节点
root->right = deleteNode(root->right, temp->data);
}
}
return root;
}
// 查找节点
TreeNode* searchNode(TreeNode* root, int data) {
if (root == NULL || root->data == data) {
return root;
} else if (data < root->data) {
// 递归在左子树中查找
return searchNode(root->left, data);
} else {
// 递归在右子树中查找
return searchNode(root->right, data);
}
}
// 查找最小节点
TreeNode* findMin(TreeNode* root) {
if (root == NULL) {
return NULL;
} else if (root->left == NULL) {
return root;
} else {
return findMin(root->left);
}
}
// 前序遍历
void preorderTraversal(TreeNode* root) {
if (root != NULL) {
printf("%d ", root->data);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
}
// 中序遍历
void inorderTraversal(TreeNode* root) {
if (root != NULL) {
inorderTraversal(root->left);
printf("%d ", root->data);
inorderTraversal(root->right);
}
}
// 后序遍历
void postorderTraversal(TreeNode* root) {
if (root != NULL) {
postorderTraversal(root->left);
postorderTraversal(root->right);
printf("%d ", root->data);
}
}
int main() {
TreeNode* root = NULL;
// 插入节点
root = insertNode(root, 50);
root = insertNode(root, 30);
root = insertNode(root, 20);
root = insertNode(root, 40);
root = insertNode(root, 70);
root = insertNode(root, 60);
root = insertNode(root, 80);
// 打印二叉树
printf("前序遍历: ");
preorderTraversal(root);
printf("\n");
printf("中序遍历: ");
inorderTraversal(root);
printf("\n");
printf("后序遍历: ");
postorderTraversal(root);
printf("\n");
// 查找节点
int searchData = 60;
TreeNode* searchResult = searchNode(root, searchData);
if (searchResult != NULL) {
printf("找到节点 %d\n", searchData);
} else {
printf("未找到节点 %d\n", searchData);
}
// 删除节点
int deleteData = 30;
root = deleteNode(root, deleteData);
printf("删除节点 %d\n", deleteData);
// 打印删除节点后的二叉树
printf("中序遍历: ");
inorderTraversal(root);
printf("\n");
return 0;
}
```
这段代码实现了二叉树的基本操作,包括创建节点、插入节点、删除节点、查找节点以及前序、中序和后序遍历等操作。你可以根据需要调用这些函数来构建和操作二叉树。
#include <stdio.h>
#include <stdlib.h>
// 二叉树节点结构
typedef struct TreeNode {
int data; // 节点数据
struct TreeNode* left; // 左子节点指针
struct TreeNode* right; // 右子节点指针
} TreeNode;
// 创建新节点
TreeNode* createNode(int data) {
TreeNode* newNode = (TreeNode*)malloc(sizeof(TreeNode));
if (newNode == NULL) {
printf("内存分配失败\n");
return NULL;
}
newNode->data = data;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
// 插入节点
TreeNode* insertNode(TreeNode* root, int data) {
if (root == NULL) {
// 如果树为空,创建一个新节点作为根节点
root = createNode(data);
} else {
if (data <= root->data) {
// 如果插入值小于等于当前节点值,递归插入左子树
root->left = insertNode(root->left, data);
} else {
// 如果插入值大于当前节点值,递归插入右子树
root->right = insertNode(root->right, data);
}
}
return root;
}
// 删除节点
TreeNode* deleteNode(TreeNode* root, int data) {
if (root == NULL) {
return root;
} else if (data < root->data) {
// 递归删除左子树中的节点
root->left = deleteNode(root->left, data);
} else if (data > root->data) {
// 递归删除右子树中的节点
root->right = deleteNode(root->right, data);
} else {
// 找到要删除的节点
if (root->left == NULL && root->right == NULL) {
// 情况1: 要删除的节点为叶子节点
free(root);
root = NULL;
} else if (root->left == NULL) {
// 情况2: 要删除的节点只有右子节点
TreeNode* temp = root;
root = root->right;
free(temp);
} else if (root->right == NULL) {
// 情况3: 要删除的节点只有左子节点
TreeNode* temp = root;
root = root->left;
free(temp);
} else {
// 情况4: 要删除的节点有左右子节点
// 找到右子树中的最小节点作为替代节点
TreeNode* temp = findMin(root->right);
root->data = temp->data;
// 递归删除右子树中的替代节点
root->right = deleteNode(root->right, temp->data);
}
}
return root;
}
// 查找节点
TreeNode* searchNode(TreeNode* root, int data) {
if (root == NULL || root->data == data) {
return root;
} else if (data < root->data) {
// 递归在左子树中查找
return searchNode(root->left, data);
} else {
// 递归在右子树中查找
return searchNode(root->right, data);
}
}
// 查找最小节点
TreeNode* findMin(TreeNode* root) {
if (root == NULL) {
return NULL;
} else if (root->left == NULL) {
return root;
} else {
return findMin(root->left);
}
}
// 前序遍历
void preorderTraversal(TreeNode* root) {
if (root != NULL) {
printf("%d ", root->data);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
}
// 中序遍历
void inorderTraversal(TreeNode* root) {
if (root != NULL) {
inorderTraversal(root->left);
printf("%d ", root->data);
inorderTraversal(root->right);
}
}
// 后序遍历
void postorderTraversal(TreeNode* root) {
if (root != NULL) {
postorderTraversal(root->left);
postorderTraversal(root->right);
printf("%d ", root->data);
}
}
int main() {
TreeNode* root = NULL;
// 插入节点
root = insertNode(root, 50);
root = insertNode(root, 30);
root = insertNode(root, 20);
root = insertNode(root, 40);
root = insertNode(root, 70);
root = insertNode(root, 60);
root = insertNode(root, 80);
// 打印二叉树
printf("前序遍历: ");
preorderTraversal(root);
printf("\n");
printf("中序遍历: ");
inorderTraversal(root);
printf("\n");
printf("后序遍历: ");
postorderTraversal(root);
printf("\n");
// 查找节点
int searchData = 60;
TreeNode* searchResult = searchNode(root, searchData);
if (searchResult != NULL) {
printf("找到节点 %d\n", searchData);
} else {
printf("未找到节点 %d\n", searchData);
}
// 删除节点
int deleteData = 30;
root = deleteNode(root, deleteData);
printf("删除节点 %d\n", deleteData);
// 打印删除节点后的二叉树
printf("中序遍历: ");
inorderTraversal(root);
printf("\n");
return 0;
}