786. 第k个数（快排）

最新推荐文章于 2023-02-06 14:52:24 发布

Syuue

最新推荐文章于 2023-02-06 14:52:24 发布

阅读量242

点赞数

分类专栏：基础算法文章标签：快速排序

本文链接：https://blog.csdn.net/weixin_43793774/article/details/107426667

版权

基础算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

AcWing 786. 第k个数

题目
思路
- 代码

题目传送门

题解思路参考大佬

题目

给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列的第k小的数是多少。

输入格式

第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在1~ $10^9$ 范围内），表示整数数列。

输出格式

输出一个整数，表示数列的第k小数。

数据范围

$,\\ 1≤k≤n$

输入输出样例

样例输入1

5 3
2 4 1 5 3

样例输出1

思路

代码

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int q[N], n, k;
void quick_sort(int q[], int l, int r) {
    if(l >= r) return;
    int x = q[l + r >> 1], i = l - 1, j = r + 1;
    while(i < j) {
        do i++; while(q[i] < x);
        do j--; while(q[j] > x);
        if(i < j) swap(q[i], q[j]);
    } 
    quick_sort(q, l, j);
    quick_sort(q, j + 1, r);
}
int main() {
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i++) scanf("%d", &q[i]);
    quick_sort(q, 0, n - 1);
    printf("%d", q[k - 1]);
    return 0;
}