AcWing 786. 第k个数
原题链接
简单
作者:
AAedion
,
2021-02-12 15:28:59
,
所有人可见
,
阅读 221
C++ 代码
#include <iostream>
using namespace std;
int n;
const int N = 1e6+10;
void quick_sort(int array[], int l, int r, int k) {
if (l >= r) {
if(l == k) {
printf("%d\n", array[l]);
}
return;
}
int i = l-1, j = r+1;
int goal = array[l];
while (i<j) {
while (array[++i] < goal);
while (array[--j] > goal);
if (i<j) {
swap(array[i], array[j]);
}
}
if(j+1 > k) {
quick_sort(array, l, j, k);
} else if (j+1 <= k) {
quick_sort(array, j+1, r, k);
}
}
int main() {
int k;
int array[N];
scanf("%d", &n);
scanf("%d", &k);
for (int i=0; i<n; i++) {
scanf("%d", &array[i]);
}
quick_sort(array, 0, n-1, k-1);
return 0;
}