【acwing】786. 第k个数

最新推荐文章于 2024-09-06 09:00:00 发布

percation

最新推荐文章于 2024-09-06 09:00:00 发布

阅读量128

点赞数

分类专栏： acwing 文章标签： c++ 算法数据结构

本文链接：https://blog.youkuaiyun.com/qq_45866781/article/details/125628686

版权

acwing 专栏收录该内容

142 篇文章

订阅专栏

本文介绍了一个基于快速排序的变种算法——快速选择算法，并详细解释了其工作原理。该算法能够在O(n)的时间复杂度内找出第k小的元素，通过不断划分数组，将查找范围缩小到特定区间。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

穿越隧道
在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int q[N];
int n,k;
int qs(int l, int r, int k){
    if(l == r) return q[l];
    int i = l - 1, j = r + 1, x = q[(l + r)/2];
    while(i < j){
        while(q[++i] < x) ;//实现左区间向右边前进，右区间向左边前进
        while(q[--j] > x) ;
        if(i < j) swap(q[i],q[j]);
    }
    int s1 = j - l  + 1;//找到当时分界点在区间的位置，因区间从0开始，所以要加1
    if(k <= s1)//当所要查询的数，在当前分界点的左边时，则进行左递归，否则进行右递归。
        return qs(l, j, k);
    return qs(j + 1,r, k - s1);
    
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i = 0; i < N; i++){
        scanf("%d",&q[i]);
    }
    cout << qs(0,n-1,k) << endl;
    return 0;
}