第k个数：快速选择

第一讲 基础算法/786.cpp

@@ -0,0 +1,45 @@
+#include <iostream>
+
+using namespace std;
+
+const int N = 100010;
+
+int q[N];
+
+int quick_sort(int q[], int l, int r, int k)
+{
+    if (l >= r)
+        return q[l]; // l=r 表示只有一个值进行排序，则直接返回这个位置的值
+
+    int i = l - 1, j = r + 1;
+    int x = q[l + r >> 1]; // 移位运算符 >> 表示 l+r 除以了 2
+    while (i < j)          // i 索引位置在 j 索引位置的前面，则循环
+    {
+        do
+            i++;
+        while (q[i] < x); // i++ 表示索引 i 向后移动一位，如果 q[i] 的值此时依然小于 x 的值，则继续向后移动索引 i，直到找到比 x 大的值
+        do
+            j--;
+        while (q[j] > x); // j-- 表示索引 j 向前移动一位，如果 q[j] 的值此时依然大于 x 的值，则继续向前移动索引 j， 知道找到比 x 小的值
+        if (i < j)
+            swap(q[i], q[j]); // 当找到了比 x 大的 q[i] ，以及比 x 小的 q[j] 值；且满足这个时候 i 依然在 j 索引的前面位置时，交换 i j 索引位置的值
+    }
+
+    // 每次都排序好一半
+    if (j - l + 1 >= k)
+        return quick_sort(q, l, j, k);
+    else
+        return quick_sort(q, j + 1, r, k - (j - l + 1));
+}
+
+int main()
+{
+    int n, k;
+    scanf("%d%d", &n, &k);
+
+    for (int i = 0; i < n; i++)
+        scanf("%d", &q[i]);
+
+    cout << quick_sort(q, 0, n - 1, k) << endl;
+    return 0;
+}

第一讲 基础算法/README.md

@@ -33,4 +33,42 @@
 
 ```
 1 2 3 4 5
-```
+```
+
+
+
+## 786. 第k个数
+
+给定一个长度为 *n* 的整数数列，以及一个整数 *k*，请用快速选择算法求出数列从小到大排序后的第 *k* 个数。
+
+#### 输入格式
+
+第一行包含两个整数 *n* 和 *k*。
+
+第二行包含 *n* 个整数（所有整数均在 1∼$10^9 $范围内），表示整数数列。
+
+#### 输出格式
+
+输出一个整数，表示数列的第 *k* 小数。
+
+#### 数据范围
+
+1≤*n*≤100000,
+
+1≤*k*≤*n*
+
+#### 输入样例：
+
+```
+5 3
+2 4 1 5 3
+```
+
+#### 输出样例：
+
+```
+3
+```
+
+
+