考研数据结构（每日一题）day48

考研数据结构（每日一题）

题目：编写一个递归算法，在一棵有n个结点的、随机建立起来的二叉排序树上查找第k(1<=k<=n)小的元素，并返回指向该结点的指针。要求算法的平均时间复杂度为O(log2n)，二叉排序树的每个结点中除data，lchild，rchild等数据成员外，增加一个count成员，保存以该结点为根的子树上的结点个数。

算法思想：

设二叉排序树的根结点为*t，根据结点存储的信息，有以下几种情况：

1. t->lchild为空时

若t->rchild非空且k==1，则*t即为第k小的元素，查找成功

若t->rchild非空且k!=1，则第k小的元素必在*t的右子树

2. t->rchild为空时

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)小的元素

完整代码：

BSTNode *Search_Small(BSTNode *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));
        }
    }
}