在这些函数的基础上,在主函数中加入1创建自定义二叉树,2查看前序遍历,3查看后序遍历,4查看后序遍历,0退出
时间: 2024-02-22 08:59:41 浏览: 48
好的,以下是添加了上述功能的示例代码:
```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);
}
if (data < root->data) {
root->left = insertNode(root->left, data);
} else if (data > root->data) {
root->right = insertNode(root->right, data);
}
return root;
}
// 二叉树前序遍历
void preOrderTraversal(TreeNode* root) {
if (root == NULL) {
return;
}
printf("%d ", root->data);
preOrderTraversal(root->left);
preOrderTraversal(root->right);
}
// 二叉树中序遍历
void inOrderTraversal(TreeNode* root) {
if (root == NULL) {
return;
}
inOrderTraversal(root->left);
printf("%d ", root->data);
inOrderTraversal(root->right);
}
// 二叉树后序遍历
void postOrderTraversal(TreeNode* root) {
if (root == NULL) {
return;
}
postOrderTraversal(root->left);
postOrderTraversal(root->right);
printf("%d ", root->data);
}
int main() {
TreeNode* root = NULL;
int choice, n, data;
while (1) {
printf("\n-------------------------\n");
printf("1. 创建二叉树\n");
printf("2. 前序遍历\n");
printf("3. 中序遍历\n");
printf("4. 后序遍历\n");
printf("0. 退出\n");
printf("-------------------------\n");
printf("请输入您的选择:");
scanf("%d", &choice);
switch (choice) {
case 1:
printf("请输入二叉树节点个数:");
scanf("%d", &n);
printf("请输入%d个节点的值:\n", n);
for (int i = 0; i < n; i++) {
scanf("%d", &data);
root = insertNode(root, data);
}
break;
case 2:
printf("二叉树前序遍历:");
preOrderTraversal(root);
printf("\n");
break;
case 3:
printf("二叉树中序遍历:");
inOrderTraversal(root);
printf("\n");
break;
case 4:
printf("二叉树后序遍历:");
postOrderTraversal(root);
printf("\n");
break;
case 0:
printf("程序已退出!\n");
exit(0);
default:
printf("无效的选择!\n");
}
}
return 0;
}
```
在这个程序中,我们添加了一个`while`循环,使程序可以一直运行,直到用户输入`0`退出。每次循环,程序会打印一个菜单,用户可以选择创建二叉树、前序遍历、中序遍历或后序遍历。
当用户选择创建二叉树时,程序会要求用户输入二叉树节点的个数和每个节点的值,并依次插入到二叉树中。
当用户选择前序遍历、中序遍历或后序遍历时,程序会分别调用对应的遍历函数,并输出结果。
当用户选择退出时,程序会退出并打印一条消息。
希望这个示例能对你有所帮助!
阅读全文