查找一个数组的K大值

时间:2022-08-05 14:46:25

前面有一篇博客,实现了如何处理寻找数组第二大数据,但是当寻找第三大、第四大时,那样的思想便存在一大缺点(if-else结构情况很多)!!!

#include <stdio.h>

// 快排将数组调整为降序数组。
// 快速排序有一个特点:每一趟一定会将一个元素放在最终位置(每一趟的low表示该确定位置元素的下标)。
int quick (int array[], int left, int right, int k)
{
    if (left <= right) {
        int key = array[low] ;
        int low = left ;
        int high = right ;
        // 快速排序核心部分
        while(low < hjgh) {
            while (low<high && array[high] <= key) {
                high-- ;
            }
            array[low] = array[high] ;

            while (low<high && array[low]>key) {
                low++ ;
            }
            array[high] = array[low] ;
        }
        array[low] = key ;
        // 一趟结束后,看看当前的low+1是否就是k值,若是,则表明第k大已经找到!!!
        // 若不是,则根据low+1与k的大小关系,确定所找元素在low的左侧还是右侧,然后递归处理即可。
        if (low == k-1) {
            //返回的是low下标所在的元素值
            return array[low] ;
        } else if (low > k-1) {
            quick (array, left, low-1, k) ;
        } else {
            quick (array, low+1, right, k) ;
        }
    }
}

// 寻找第K大元素,利用快排算法。
int findK (int array[], int n, int k)
{
    return quick (array, 0, n-1, k) ;
}

// 主函数
int main (void)
{   
    int array[5] = {1,3,2,4,2} ;

    printf ("%d\n", findK(array, 5, 3)) ;

    return 0 ;
}