BSTNode* createBST(BSTNode* root, int data) { if (root == NULL) { root = (BSTNode*)malloc(sizeof(BSTNode)); root->data = data; root->left = NULL; root->right = NULL; } else if (data < root->data) { root->left = createBST(root->left, data); } else if (data > root->data) { root->right = createBST(root->right, data); } return root; }解释代码
时间: 2024-04-21 16:28:47 浏览: 118
这段代码是一个递归函数,用于向一个二叉搜索树中插入一个新节点。函数的参数是二叉搜索树的根节点指针和要插入的节点数据。函数首先判断二叉搜索树是否为空,如果是,则创建一个新节点,并将数据存储在该节点中,并将左右子节点指针初始化为NULL。如果二叉搜索树不为空,则比较要插入的数据和当前节点的数据大小,如果小于当前节点的数据,则递归调用函数并将当前节点的左子节点作为根节点,将要插入的数据作为参数传递进去;如果大于当前节点的数据,则递归调用函数并将当前节点的右子节点作为根节点,将要插入的数据作为参数传递进去。最后,函数返回二叉搜索树的根节点指针。
相关问题
优化这段代码:#include<stdio.h> #include<assert.h> #include<string.h> #include<errno.h> #include<stdlib.h> typedef int KeyType; typedef char InfoType; typedef struct node { KeyType key; InfoType data; struct node*lchild,*rchild; }BSTNode; BSTNode *insertbst(BSTNode*&bt,KeyType k) { if(bt==NULL) { bt=(BSTNode*)malloc(sizeof(BSTNode)); bt->key=k;bt->lchild=bt->rchild=NULL; } else if(k<bt->key) bt->lchild=insertbst(bt->lchild,k); else if(k>bt->key) bt->rchild=insertbst(bt->rchild,k); return bt; } BSTNode*createbst(KeyType a[],int n) { BSTNode*bt=NULL; int i=0; while(i<n) { bt=insertbst(bt,a[i]); i++; } return bt; } BSTNode *searchbst(BSTNode*bt,KeyType k) { BSTNode *p=bt; while(p!=NULL) { if(p->key==k)break; else if(p->key<k) p=p->lchild; else p=p->rchild; } return p; } void dispbst(BSTNode*bt) { if(bt!=NULL) { printf("%d",bt->key); if(bt->lchild!=NULL||bt->rchild!=NULL) { printf("("); dispbst(bt->lchild); if(bt->rchild!=NULL) printf(","); dispbst(bt->rchild); printf(")"); } } } int main() { BSTNode *bt,q; KeyType k=6; int a[]={4,9,0,1,8,6,3,5,2,7}; int n=10; bt=createbst(a,n); printf("输出这棵树"); dispbst(bt); printf("\n要查找的关键字为:%d",k); q=searchbst(bt,k); if((q!=0) printf("输出关键字6所在的位置:%d\n",q); else printf("\n关键字6的元素不在表中\n"); return 1; }
以下是对代码的优化:
1. 将代码中的Keytype和InfoType定义提前到头文件中,以便在多个文件中使用。
2. 在insertbst函数中,使用了递归的方式实现插入,但是可以使用迭代的方式来实现,代码更加简洁。
3. 在createbst函数中,可以使用for循环替代while循环,代码更加简洁。
4. 在searchbst函数中,如果找到了目标节点,应该返回指向该节点的指针,否则应该返回NULL。
5. 在输出函数dispbst中,应该先输出左子树,再输出右子树。
优化后的代码如下:
```c
#include <stdio.h>
#include <stdlib.h>
typedef int KeyType;
typedef char InfoType;
typedef struct node {
KeyType key;
InfoType data;
struct node *lchild, *rchild;
} BSTNode;
BSTNode *insertbst(BSTNode *&bt, KeyType k) {
BSTNode *p = bt, *prev = NULL;
while (p != NULL) {
prev = p;
if (k < p->key) {
p = p->lchild;
} else if (k > p->key) {
p = p->rchild;
} else {
return bt;
}
}
p = (BSTNode *) malloc(sizeof(BSTNode));
p->key = k;
p->lchild = p->rchild = NULL;
if (prev == NULL) {
bt = p;
} else if (k < prev->key) {
prev->lchild = p;
} else {
prev->rchild = p;
}
return bt;
}
BSTNode *createbst(KeyType a[], int n) {
BSTNode *bt = NULL;
for (int i = 0; i < n; i++) {
insertbst(bt, a[i]);
}
return bt;
}
BSTNode *searchbst(BSTNode *bt, KeyType k) {
BSTNode *p = bt;
while (p != NULL && p->key != k) {
if (k < p->key) {
p = p->lchild;
} else {
p = p->rchild;
}
}
return p;
}
void dispbst(BSTNode *bt) {
if (bt != NULL) {
printf("%d", bt->key);
if (bt->lchild != NULL || bt->rchild != NULL) {
printf("(");
dispbst(bt->lchild);
if (bt->rchild != NULL) {
printf(",");
}
dispbst(bt->rchild);
printf(")");
}
}
}
int main() {
BSTNode *bt, *q;
KeyType k = 6;
int a[] = {4, 9, 0, 1, 8, 6, 3, 5, 2, 7};
int n = 10;
bt = createbst(a, n);
printf("输出这棵树:");
dispbst(bt);
printf("\n要查找的关键字为:%d", k);
q = searchbst(bt, k);
if (q != NULL) {
printf("\n输出关键字6所在的位置:%d\n", q->key);
} else {
printf("\n关键字6的元素不在表中\n");
}
return 1;
}
```
经过优化后,代码更加简洁易读,并且运行效率也有所提高。
#include <stdio.h> #include <malloc.h> typedef int TElemType; typedef int Status; #define OK 1; #define ERROR -1; typedef struct BSTNode{ TElemType data; struct BSTNode *leftchild,*rightchild; }BSTNode,*BSTree; Status CreateBST(BSTree &T,int x) { if(T==NULL) { T = (BSTree)malloc(sizeof(BSTNode)); T->data=x; T->leftchild=T->rightchild=NULL; return OK; } if (x < T->data) { CreateBST(T->leftchild, x); } else { CreateBST(T->rightchild, x); } return OK; } Status print(BSTree T) { if(T) { print(T->leftchild); printf("%d ",T->data); print(T->rightchild); } return OK; } int main() { int a[100]; int n,i; BSTree T=NULL; scanf("%d",&n); for(i=0;i<n;i++) { scanf("%d",&a[i]); } for(i=0;i<n;i++) { CreateBST(T,a[i]); } print(T); printf("\n"); return 0; }
这段代码实现了二叉搜索树的创建和遍历操作。具体来说,它通过递归的方式将一个给定的数组中的元素插入到二叉搜索树中,并最终以中序遍历的方式输出整个二叉搜索树的节点值。
其中,结构体 BSTNode 定义了每个节点的数据类型 TElemType 和指向左右子节点的指针 leftchild 和 rightchild。函数 CreateBST 实现了二叉搜索树的创建,它首先判断当前节点是否为空,如果是则将该节点的值设为给定值 x,并将左右子节点指针设为 NULL;否则,如果 x 小于当前节点的值,则递归处理左子树,否则递归处理右子树。函数 print 以递归方式实现了二叉搜索树的中序遍历,即先输出左子树,再输出当前节点的值,最后输出右子树。
在 main 函数中,首先读入一个整数 n 表示数组元素个数,然后读入 n 个整数作为数组元素,最后将这些元素插入到二叉搜索树中并输出。
阅读全文