设根结点为*t,有以下几种情况:
1、t->lchild为空时:
- 若t->rchild非空且k==1,则*t即为第k小元素,查找成功。
- 若t->rchild非空且k!=1,则第k小元素必在*t的右子树中。
2、t->lchild非空时:
- t->lchild->count==k-1,*t即为第k小的元素。
- t->lchild->count>k-1,第k小的元素必在*t的左子树,继续到*t的左子树中查找。
- t->lchild->count<k-1,第k小的元素必在右子树,继续搜索右子树,寻找第k-(t->lchild->count+1)小的元素。
typedef struct BSTNode{
struct BSTNode *lchild,rchild;
int data,count;
}*BSTree;
BSTNode *Search_Small(BSTree t,int k){
if(k<1||k>t->count) return NULL;
if(t->lchild==NULL){
if(k==1) return t;
else return Search_Small(t->rchild,k-1);
}
else{
if(t->lchild->count==k-1) return t;
if(t->lchild->count>k-1) return Search_Small(t->lchild,k);
if(t->lchild->count<k-1) return Search_Small(t->rchild,k-(t->lchild->count+1));
}
}