给定一个无序数组，求第K大的数，要求不能使用额外的空间，时间复杂度尽量低

今天快手面试，面试官问我对各种排序算法了解吗，我说了解，然后他给我出了一道题，给定一个无序数组，求第K大的数，要求不能使用额外的空间，时间复杂度尽量低，当时有点懵…我能想到的只能是用数组滚动遍历一遍，时间复杂度为O(n),空间复杂度也O(n)…或者排序再遍历，实在想不出，面试官给了一个思路：
使用分治和快排的思想，这才恍然大悟。

寻找过程：
1.用快排的思想寻找基准点，比基准大的元素数目标志位m正好为k-1，基准就是目的元素。
2.比基准大的元素标志位m小于k, 那么就在比基准小的集合里面找第(k-m)大的元素
3.若是比基准大的元素为m大于k，那就继续在该集合里面找第k大的元素。

贴上代码：

#include <bits/stdc++.h>
using namespace std;
int findK(int* arr, int left, int right, int k) {
    if(left >= right) return arr[left];
    int first = left, last = right;
    int key = arr[first];
    while(first < last) {
        while(first < last && arr[last] >= key) last--;
        arr[first] = arr[last];
        while(first < last && arr[first] <= key) first++;
        arr[last] = arr[first];
    }
    arr[first] = key;
    if(first == k) return arr[k];
    else {
        if(first > k) return findK(left, first, arr, k);
        else return findK(first+1, right, arr, k);
    }
}
int main() {
    int n;
    int arr[1000];
    cin>>n;
    for(int i = 1; i <= n; i++) cin >>arr[i];
    int k;
    cin >> k;
    int num = findK(1, n, arr, k);
    cout << num << endl;
    return 0;
}