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

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

题目：编写一个算法，使之能够在数组L[1…n]中找出第k小的元素（即从小到大排序后处于第k个位置的元素）

算法思想：

• 首先本题可以采取排序算法对数组进行从小大的排序，然后提取L(k)，如此便得到了第k小元素，但是平均时间复杂度为O(nlog2n)以上。

• 还可采用小顶堆的方法，每次堆顶的元素都是最小值元素，时间复杂度为O(n+klog2n)

• 以下详细写一下关于快速排序的操作方法：

• 从数组L[1…n]中选择一个轴pivot（随机或者取第一个）进行和快速排序一样的划分操作后，表L[1…n]被划分为L[1…m-1]和L[m+1…n]，其中L(m)=pivot

• 当m=k时，pivot就是所要的元素，直接返回pivot即可

• 当m<k时，所找的元素一定在L[m+1…n]中，因此可对L[m+1…n]递归查找第k-m小的元素

• 当m>k时，所找的元素一定在L[1…m-1]中，因此可对L[1…m-1]递归查找第k小的元素

完整代码：

int kth_elem(int a[],int low,int high,int k){
    int pivot = a[low];
    int low_temp = low;
    int high_temp = high;
    //快排划分算法
    while (low < high)
    {
        while (low < high && a[high] >= pivot)
        {
            --high;
        }
        a[low] = a[high];
        while (low < high && a[low] <= pivot)
        {
            ++low;
        }
        a[high] = a[low];
    }
    a[low] = pivot;
    if (low == k)   //与k相同，直接返回pivot元素
    {
        return a[low];
    }
    else if (low > k)   //在前一部分表中递归
    {
        return kth_elem(a,low_temp,low - 1,k);
    }
    else{     //在后一部分递归
        return kth_elem(a,low + 1,high_temp,k);
    }
}