[编程题]整数无序数组求第K大数

关键词：TopK问题，堆，随机选择

试题链接：
https://www.nowcoder.com/questionTerminal/097ab63cffa847d89716f2ca8c23524f
问题描述：

这道题我使用堆把它搞定了，之后会继续整理随机选择算法的解决方案。

参考思路3：
利用堆的解决方案：

#include<iostream>
#include<vector>
#include<queue>

using namespace std;

int main(){
    vector<int> a;
    int x;
    while(cin>>x){
        a.push_back(x);
    }
    priority_queue<int,vector<int>,greater<int>>q;
    int i,j;
    for(i=0;i<a.back();i++){
        q.push(a[i]);
    }
    for(j=i;j<a.size()-1;j++){
        if(a[j]>q.top()){//小于存储前k大的优先队列中最小的元素
            q.pop();//pop掉最小的元素
            q.push(a[j]); //放进当前的元素
        } 

    }
    cout << q.top(); //重复以上操作后top处就是第k大
}

参考思路4：线性时间复杂度
利用随机选择算法的解决方案：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<algorithm>
using namespace std;

int partition(int arr[], int low, int high) {
    //生成low到high之间的随机数
    int p = (int(1.0*rand() /RAND_MAX * (high - low))+low);
    //交换数组中p和low
    swap(arr[p], arr[low]);
    int tmp = arr[low];
    //交换数组中左边第一个比tmp大的数和右边第一个比tmp小的数
    while (low < high) {
        while (low<high && arr[high]>tmp) {
            high--;
        }
        arr[high] = arr[low];
        while (low < high && arr[low] <= tmp) {
            low++;
        }
        arr[low] = arr[high];
    }
    arr[low] = tmp; //tmp位于high和low相遇的地方
    return low; //返回相遇的位置
}

//随机选择算法
int randSelect(int arr[], int low, int high, int k) {
    if (low == high)return arr[low]; //到达边界
    int i = partition(arr, low, high);//划分后主元位置为i
    int j = i - low;//i为low与high之间的第i-low+1大
    if (j == k) {
        return arr[i];
    }
    else if (j > k) {
        return randSelect(arr, low, i - 1, k);
    }
    else {
        return randSelect(arr, i + 1, high, k - j);
    }
}

int main() {
    int arr[10000];
    int i = 0;
    int x;
    while (true) {
        cin >>x; 
        arr[i++]=x;
        if (getchar() == '\n')
            break;
    }
    int n = i - 1;
    int k = arr[i - 1];
    cout << randSelect(arr, 0, n - 1, k);

}

以上代码在本地测试没有毛病，

然而在牛客网上测试时出现了错误：

查找了一些解决方案，参考博客https://blog.csdn.net/sajiazaici/article/details/78066925试着解决了一下，然鹅仍然没有解决问题，我就决定不在这个上面过多纠结了orz（输入也奇怪，又报这种错误= =有纠结的时间还不如多看看随机选择算法的细节部分）。