寻找第K个数字

大致题意: 给出一个含有任意数字的序列, 找出其中升序排列下第k个数字.

升序排列下第k个数字, 可以视为在数组中第k - 1个位置上的值. 使用快速排序的思想, 针对每轮快速排序的子过程, 判断k是在左区间还是在右区间, 递归到最后时, 只会剩下一个位置就是第k个数字.

#include <iostream>
#include <vector>

using namespace std;

vector<int> nums;

int find_k(vector<int>& arr, int left, int right, int k) {
    if (left >= right) {
        return nums[k];
    }
    int x = arr[left + (right - left) / 2];
    int i = left - 1, j = right + 1;
    while (i < j) {
        do {
            i++;
        } while (arr[i] < x);
        do {
            j--;
        } while (arr[j] > x);
        if (i < j) {
            swap(arr[i], arr[j]);
        }
    }
    if (k <= j) {
        return find_k(arr, left, j, k);
    } else {
        return find_k(arr, j + 1, right, k);
    }
}

int main(int argc, char* argv[])
{
    int n, k;
    cin >> n >> k;
    nums.resize(n);
    for (int i = 0; i < n; ++i) {
        cin >> nums[i];
    }
    cout << find_k(nums, 0, n - 1, k - 1) << endl;
    return 0;
}