数据结构二叉树实验代码
时间: 2023-06-30 11:22:46 浏览: 40
好的,我可以提供一份二叉树实验代码的示例,供你参考。以下是一个简单的二叉树实现,包括节点的结构体定义、创建、遍历、插入和删除操作等。
```
#include <stdio.h>
#include <stdlib.h>
// 二叉树节点结构体定义
typedef struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
} TreeNode;
// 创建二叉树
TreeNode *createTree() {
TreeNode *root = NULL;
int val;
printf("请输入二叉树的根节点值:");
scanf("%d", &val);
if (val == -1) { // 输入-1表示结束
return NULL;
}
root = (TreeNode *)malloc(sizeof(TreeNode));
root->val = val;
root->left = createTree(); // 递归创建左子树
root->right = createTree(); // 递归创建右子树
return root;
}
// 前序遍历二叉树
void preorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
printf("%d ", root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
}
// 中序遍历二叉树
void inorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
inorderTraversal(root->left);
printf("%d ", root->val);
inorderTraversal(root->right);
}
// 后序遍历二叉树
void postorderTraversal(TreeNode *root) {
if (root == NULL) {
return;
}
postorderTraversal(root->left);
postorderTraversal(root->right);
printf("%d ", root->val);
}
// 插入节点
void insertNode(TreeNode *root, int val) {
if (root == NULL) {
return;
}
if (val < root->val) { // 插入左子树
if (root->left == NULL) {
TreeNode *node = (TreeNode *)malloc(sizeof(TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->left = node;
} else {
insertNode(root->left, val);
}
} else { // 插入右子树
if (root->right == NULL) {
TreeNode *node = (TreeNode *)malloc(sizeof(TreeNode));
node->val = val;
node->left = NULL;
node->right = NULL;
root->right = node;
} else {
insertNode(root->right, val);
}
}
}
// 删除节点
TreeNode *deleteNode(TreeNode *root, int val) {
if (root == NULL) {
return NULL;
}
if (val < root->val) { // 在左子树中删除
root->left = deleteNode(root->left, val);
} else if (val > root->val) { // 在右子树中删除
root->right = deleteNode(root->right, val);
} else { // 找到了要删除的节点
if (root->left == NULL) { // 只有右子树
TreeNode *temp = root->right;
free(root);
return temp;
} else if (root->right == NULL) { // 只有左子树
TreeNode *temp = root->left;
free(root);
return temp;
} else { // 左右子树都存在
TreeNode *temp = root->right;
while (temp->left != NULL) {
temp = temp->left;
}
root->val = temp->val;
root->right = deleteNode(root->right, temp->val);
}
}
return root;
}
// 主函数
int main() {
TreeNode *root = NULL;
int val;
root = createTree(); // 创建二叉树
printf("前序遍历:");
preorderTraversal(root); // 前序遍历
printf("\n中序遍历:");
inorderTraversal(root); // 中序遍历
printf("\n后序遍历:");
postorderTraversal(root); // 后序遍历
printf("\n请输入要插入的节点值:");
scanf("%d", &val);
insertNode(root, val); // 插入节点
printf("插入后的中序遍历:");
inorderTraversal(root); // 中序遍历
printf("\n请输入要删除的节点值:");
scanf("%d", &val);
root = deleteNode(root, val); // 删除节点
printf("删除后的中序遍历:");
inorderTraversal(root); // 中序遍历
return 0;
}
```
以上是一个简单的二叉树实现示例,你可以根据自己的需求进行修改和扩展。
相关推荐















