diff --git a/第一讲 基础算法/786.cpp b/第一讲 基础算法/786.cpp new file mode 100644 index 0000000..f6528e4 --- /dev/null +++ b/第一讲 基础算法/786.cpp @@ -0,0 +1,45 @@ +#include + +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; +} diff --git a/第一讲 基础算法/README.md b/第一讲 基础算法/README.md index 8888a53..6f424e4 100644 --- a/第一讲 基础算法/README.md +++ b/第一讲 基础算法/README.md @@ -33,4 +33,42 @@ ``` 1 2 3 4 5 -``` \ No newline at end of file +``` + + + +## 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 +``` + + +