快速选择算法

快速选择算法

算法思想：

1. 随机选择一个枢轴。
2. 使用划分算法将枢轴放在数组中的合适位置 pos。将小于枢轴的元素移到左边，大于等于枢轴的元素移到右边。
3. 比较 pos 和 N - k 以决定在哪边继续递归处理。

int partition(vector<int>& nums, int l, int r)
{
    int left = l+1, right = r;
    swap(nums[l], nums[l+r>>1]);
    int bound = nums[l];

    while (left <= right)
    {
        while (left <= right && nums[left] > bound) ++left;
        while (left <= right && nums[right] < bound) --right;
        if (left < right) swap(nums[left++], nums[right--]);
        else break;
    }

    swap(nums[right], nums[l]);
    return right;
}

int findKthLargest(vector<int>& nums, int k) {
    int n = nums.size();

    int l = 0, r = n-1;
    while (true)
    {
        int pos = partition(nums, l, r);
        if (pos == k-1) return nums[pos];
        else if (pos < k-1) l = pos+1;
        else r = pos-1;
    }

    return -1;
}