第K个数

一、题目描述
给定一个长度为 nn 的整数数列,以及一个整数 kk,请用快速选择算法求出数列从小到大排序 后的第 kk 个数。
输入格式
第一行包含两个整数 nn 和 kk。
第二行包含 nn 个整数(所有整数均在 1∼1091∼109 范围内),表示整数数列。
输出格式
输出一个整数,表示数列的第 kk 小数。
数据范围
1≤n≤100000,1≤k≤n1≤n≤100000,1≤k≤n
输入样例:

5 3 2 4 1 5 3 

输出样例:

3 

二、理解感悟

1、这是快速排序模板的练习题。

2、不一样的地方在于它可以利用快排模板,但却不需要真的把所有数据排序完成,每次一分为二后,只关心自己所有的那一半,就是可以节约一半的递归时间。

3、由于是关心 位置(第几个),所以在递归时需要携带这个参数。

4、位置这个参数 不是一成不变的,因为如果在左侧,那么就是原来的位置,如果在右侧,那就需要减去整个左侧的长度。这个 kk参数可以理解为 在当前数组中的位置,最终将确定这个位置上的值。

5、最后,直接使用q[k]q[k]就是拿到了最终这个位置上应该存在的值。

三、实现代码

#include <bits/stdc++.h>

using namespace std;
const int N = 100010;
int q[N];
int n, k;

void quick_sort(int q[], int l, int r, int k) {
    if (l >= r) return;
    int i = l - 1, j = r + 1, x = q[(l + 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]);
    }
    int len = j - l + 1; // 左侧长度
    if (k <= len)
        quick_sort(q, l, j, k); // 左侧
    else
        quick_sort(q, j + 1, r, k - len); // 右侧
}

int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) cin >> q[i];
    quick_sort(q, 1, n, k);
    printf("%d", q[k]);
    return 0;
} 

全部评论

相关推荐

在写周报的打工人很独...:这个笔试昨天晚上做了一下,真难啊,前后端,ai全有
点赞 评论 收藏
分享
我要娶个什么名:学长你电脑闹鬼了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务