寻找第k个最大数 C++

题目描述
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。

1.暴力法

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        sort(begin(a),end(a));
        return a[n-K];
    }

2.快排法

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        return quickSortK(a,0,n-1,K);
        
    }
    int quickSortK(vector<int>&nums,int start,int end,int K){
        int index=partition(nums,start,end);
        if(index==K-1) return nums[index];
        if(index<K-1) return quickSortK(nums,index+1,end,K);
        return quickSortK(nums,start,index-1,K);
    }
    int partition(vector<int>&nums,int start,int end){//引用传递
        int k=rand()%(end-start+1)+start;//随机数增速
        swap(nums[k],nums[end]);
        int i=start-1;
        for(int j=start;j<end;j++){
            if(nums[j]>=nums[end]){
                swap(nums[++i],nums[j]);
            }
        }
        i++;
        swap(nums[i],nums[end]);
        return i;
    }
};

3.堆排序法
1.自下向上构建最大堆,构建完成后堆顶为最大值
2.循环操作:删除堆顶并重新堆排序,删除操作执行K-1次后堆顶即为所求值

class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        int Size=a.size();
        buildHeap(a,Size);
        while(--K){
            swap(a[0],a[Size-1]);
            Size--;
            maxheap(a,0,Size);
        }
        return a[0];
    }
    void maxheap(vector<int>&a,int i,int Size){
        int l=2*i+1,r=2*i+2;
        int largest=i;
        if(l<Size&&a[l]>a[largest])largest=l;
        if(r<Size&&a[r]>a[largest])largest=r;
        if(largest!=i){
            swap(a[i],a[largest]);
            maxheap(a,largest,Size);
        }
    }
    
    void buildHeap(vector<int>&a,int Size){
        for(int i=Size/2;i>=0;i--){
            maxheap(a,i,Size);
        }
    }
};
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值