线性时间选择(快速选择排序)

图解：

代码如下：

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

// 交换两个元素
void swap(int& a, int& b) {
    int temp = a;
    a = b;
    b = temp;
}

// 对数组进行划分，并返回基准位置
int partition(vector<int>& arr, int left, int right) {
    // 随机选择一个基准
    int pivot_index = rand() % (right - left + 1) + left;
    // 将基准放到最右边
    swap(arr[pivot_index], arr[right]);
    // 初始化 i 和 j 指针
    int i = left; // i 指向小于等于区域的最后一个元素
    int j = left; // j 指向未处理区域的第一个元素
    // 遍历未处理区域
    while (j < right) {
        // 如果当前元素小于等于基准，就将其放到小于等于区域，并更新 i 和 j
        if (arr[j] <= arr[right]) {
            swap(arr[i], arr[j]);
            i++;
            j++;
        }
        else {
            // 否则就跳过当前元素，并更新 j
            j++;
        }
    }
    // 最后将基准放到正确位置，并返回其索引
    swap(arr[i], arr[right]);
    return i;
}

// 快速选择算法，返回数组中第 k 小（k >= 1） 的元素
int quick_select(vector<int>& arr, int left, int right, int k) {
    if (left == right) { // 只有一个元素时直接返回
        return arr[left];
    }
    
   // 对数组进行划分，并得到基准位置 
   int pivot_index = partition(arr, left, right);
    
     if (pivot_index == k - 1) { // 基准位置刚好等于 k-1，则返回其值 
         return arr[pivot_index];
     }
     else if (pivot_index > k - 1) { // 基准位置大于 k-1，则在左半部分继续查找 
         return quick_select(arr, left, pivot_index - 1, k);
     }
     else { // 基准位置小于 k-1，则在右半部分继续查找 
         return quick_select(arr, pivot_index + 1, right, k);
     }
}

int main() {
    
   vector<int> arr; 
   int n; // 数组大小 
   cin >> n; 
   int k; // 第 k 小的元素 
   cin >> k; 
   
   for (int i = 0; i < n; i++) { 
       int x; // 输入每个元素的值 
       cin >> x; 
       arr.push_back(x); // 将元素添加到数组中 
   } 
    
   cout << quick_select(arr, 0 , n - 1 ,k); 
    
}

知识点：

（1）rand() % (right - left + 1) + left;

我可以举一个简单的例子来说明这个代码的作用。假设我们想要生成一个在 10 到 20 之间的随机数，那么我们可以这样写：

int left = 10; int right = 20; int random = rand() % (right - left + 1) + left;

首先，rand() 函数会返回一个大于等于 0 且小于 RAND_MAX 的整数，比如说是 1 2 3 4 5 6 7 8 9。然后，我们用这个整数对 right - left + 1 取余，得到一个在 0 到 right - left + 1 = 11之间的整数，比如说是 7。最后，我们再加上 left ，得到一个在 left 到 right =10 到20之间的整数，比如说是17。这样就得到了我们想要的随机数。

（2）