用c语言写出二叉树的代码
时间: 2023-11-20 14:07:29 浏览: 76
以下是一个简单的二叉树结构体和一些基本操作的代码:
```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));
newNode->data = data;
newNode->left = NULL;
newNode->right = NULL;
return newNode;
}
// 插入一个节点到二叉树中
TreeNode* insertNode(TreeNode *root, int data) {
if (root == NULL) {
return createNode(data);
} else if (data <= root->data) {
root->left = insertNode(root->left, data);
} else {
root->right = insertNode(root->right, 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* 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) { // 没有子节点
free(root);
root = NULL;
} else if (root->left == NULL) { // 只有右子节点
TreeNode *temp = root;
root = root->right;
free(temp);
} else if (root->right == NULL) { // 只有左子节点
TreeNode *temp = root;
root = root->left;
free(temp);
} else { // 有两个子节点
TreeNode *temp = root->right;
while (temp->left != NULL) {
temp = temp->left;
}
root->data = temp->data;
root->right = deleteNode(root->right, temp->data);
}
}
return root;
}
// 中序遍历(左 -> 根 -> 右)
void inorderTraversal(TreeNode *root) {
if (root != NULL) {
inorderTraversal(root->left);
printf("%d ", root->data);
inorderTraversal(root->right);
}
}
int main() {
TreeNode *root = NULL;
root = insertNode(root, 5);
root = insertNode(root, 3);
root = insertNode(root, 8);
root = insertNode(root, 1);
root = insertNode(root, 4);
root = insertNode(root, 7);
root = insertNode(root, 9);
printf("中序遍历结果:");
inorderTraversal(root);
printf("\n");
root = deleteNode(root, 3);
root = deleteNode(root, 8);
printf("中序遍历结果:");
inorderTraversal(root);
printf("\n");
return 0;
}
```
这个代码演示了如何创建、插入、查找、删除一个二叉树节点,以及如何进行中序遍历。在这个例子中,我们创建了一个包含 7 个节点的二叉树,然后删除了两个节点(3 和 8),最后再次进行中序遍历,输出结果为 1 4 5 7 9。
阅读全文