在数组中找到第k大的元素(快速排序)

快速排序思想:先在数组中选择一个数,再把数组中的数字分成两部分,比选择数字小的放在左边,比选择数字大的放在右边。
这里写图片描述
代码:取数组最右边的数字作为选择的数字,从数组的第一个数字开始与选择数字进行比较,将小于选择数字放入数组左边,当挑选出所有小于选择数子的数时,将选择数字放入中间。则数组剩余的右边部分为大于等于选择数字的所有数。

int partition(int data[],int length,int start,int end){
    if(data==NULL||length<=0||start<0||end>=length)
        throw new std::exceotion("invalid parameters");

    int small=start;
    for(int i=start;i<end;i++){
        if(data[i]<data[end]){
            std::swap(data[small],data[i]);
            small++;
        }
    }
    std::swap(data[small],data[end]);
    return small;
}

这里写图片描述
递归的思想在每次选中的数字左右两边排序(递归:调用自身)
分别将得到所有小于选择数字的部分进行排序(start——index-1),将所有大于选择数字的部分进行排序(index+1——end)。(选择数字比左半部分大,比右半部分小所以递归时不需要将它进行排序)
代码:

void quicksort(int data[],int length,int start,int end){
    if(start==end)
        return;
    int index=partition(data,length,start,end);
    if(index>start)
        quicksort(data,length,start,index-1);//小于选择数字的部分进行排序
    if(index<end)
        quicksort(data,length,index+1,end);//大于选择数字的部分进行排序
}

在数组中找到第k大的元素
以某一个元素为pivot元素,将元素分为两个集合,一个集合元素比pivot小,另一个比pivot大
1.若比pivot大的元素数目正好为k-1,那么pivot就是我们要找到元素
2.若比pivot大的元素为m(小于k), 那么就在比pivot小的集合里面找第(k-m)大的元素
3.若是比pivot大的元素为m(大于k),那就继续在该集合里面找第k大的元素

 int findKthEle(vector<int>& nums, int start, int end,int k)
    {
        int index = partition(nums,start,end);
        int length = end-index+1;
        if(length == k)
            return nums[index];
        else if(length > k)
            return findKthEle(nums,index+1,end,k);
        else
            return findKthEle(nums,start,index-1,k-length);
    }

猜你喜欢

转载自blog.csdn.net/htt789/article/details/80015545