#include <stdio.h>
#include <stdlib.h>
// 定义树节点结构体
typedef struct TreeNode {
int val; // 节点的值
struct TreeNode *left; // 左子节点指针
struct TreeNode *right; // 右子节点指针
} TreeNode;
// 插入操作,root参数为二叉排序树的根节点,val为要插入的值
void insert(TreeNode **root, int val) {
// 如果根节点为空,则新建一个节点并将其设为根节点
if (*root == NULL) {
*root = (TreeNode *)malloc(sizeof(TreeNode));
(*root)->val = val;
(*root)->left = NULL;
(*root)->right = NULL;
return;
}
// 如果要插入的值小于当前节点的值,则向左子树递归插入
if (val < (*root)->val) {
insert(&((*root)->left), val);
} else { // 否则向右子树递归插入
insert(&((*root)->right), val);
}
}
// 查找操作,root参数为二叉排序树的根节点,val为要查找的值
// 如果查找到了,则返回1,否则返回0
int search(TreeNode *root, int val) {
// 如果当前节点为空,则表示查找失败
if (root == NULL) {
return 0;
}
// 如果当前节点的值等于要查找的值,则查找成功
if (root->val == val) {
return 1;
}
// 如果要查找的值小于当前节点的值,则在左子树中递归查找
if (val < root->val) {
return search(root->left, val);
} else { // 否则在右子树中递归查找
return search(root->right, val);
}
}
int main() {
TreeNode *root = NULL; // 初始化根节点为空
int n, val, x;
printf("输入数据的个数:");
scanf("%d", &n);
printf("输入 %d 个数据:", n);
for (int i = 0; i < n; i++) {
scanf("%d", &val); // 输入数据
insert(&root, val); // 将数据插入二叉排序树中
}
printf("输入要查找的数:");
scanf("%d", &x);
if (search(root, x)) { // 查找要查找的数是否在二叉排序树中
printf("%d 存在于二叉排序树中。\n", x);
} else {
printf("%d 不存在于二叉排序树中。\n", x);
}
return 0;
}