欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

查找第K大的数

程序员文章站 2022-05-12 10:57:56
...

类快排 第一种方法 o(n)

2019.06.08更正错误

#include <bits/stdc++.h>
using namespace std;

const int N = 1000;

int s[N];

int partion(int l,int r) {
    int tmp = s[l];
    int i = l, j = r;
    while(i < j) {
        while(i < j && s[j] >= tmp) j--;
        while(i < j && s[i] <= tmp) i++;
        if(i < j) {
            swap(s[i], s[j]);
        }
    }
    // now i == j
    swap(s[i], s[l]);
    return i;
}


void selectKth(int l,int r,int k) {
    if(l > r)
        return ;
    int c = partion(l,r);
    if(c == k)
        return ;
    else if(c < k)
        selectKth(c+1,r,k);
    else
        selectKth(l,c,k);
}

int main()
{
    srand(time(NULL));
    int n = 10;
    int m = 10;

    for(int i=1; i<=n; i++)
        s[i] = rand() % 15;
    int k = 5;
    selectKth(1,n,k);
    printf("%d\n", s[k]);
    sort(s+1,s+11);
    printf("%d\n", s[k]);

    return 0;
}

第二种方法 用到堆了

大根堆 是arr[i] <= arr[2*i+1] && arr[i] <= arr[2*i], 因此大根堆的堆顶是最小值,所以排序的话呢,就是从小到大排的...

priority_queue<int> que;
int k = 5;
for(int i=1; i<=n; i++) {
    que.push(s[i]);
    if(que.size() > k) {
        que.pop();
    }
}
cout<<que.top()<<endl;